Searched refs:TensorFlowRegistryInterface (Results 1 – 7 of 7) sorted by relevance
60 EXPECT_FALSE(dyn_cast<TensorFlowRegistryInterface>(*func_op)); in TEST()61 EXPECT_FALSE(dyn_cast<TensorFlowRegistryInterface>(*ret_op)); in TEST()79 auto iface = dyn_cast<TensorFlowRegistryInterface>(add); in TEST()106 auto iface = dyn_cast<TensorFlowRegistryInterface>(case_op); in TEST()
30 TEST(TensorFlowRegistryInterface, TestDefaultImplementation) { in TEST() argument40 auto iface = dyn_cast<TensorFlowRegistryInterface>(op); in TEST()70 auto iface = dyn_cast<TensorFlowRegistryInterface>(op); in TEST()
21 def TensorFlowRegistryInterface22 : OpInterface<"TensorFlowRegistryInterface"> {
50 auto registry = dyn_cast<TensorFlowRegistryInterface>(op); in getEffects()
34 : public TensorFlowRegistryInterface::FallbackModel<
155 if (interface == TypeID::get<TensorFlowRegistryInterface>()) { in getRegisteredInterfaceForOp()
59 if (auto registry = dyn_cast<TensorFlowRegistryInterface>(op)) in IsStateless()