Searched refs:TensorFlowRegistryInterfaceBase (Results 1 – 4 of 4) sorted by relevance
33 class TensorFlowRegistryInterfaceBase35 TensorFlowRegistryInterfaceBase>,36 public DialectInterface::Base<TensorFlowRegistryInterfaceBase> {38 explicit TensorFlowRegistryInterfaceBase(Dialect *dialect) in TensorFlowRegistryInterfaceBase() function39 : DialectInterface::Base<TensorFlowRegistryInterfaceBase>(dialect) {} in TensorFlowRegistryInterfaceBase()
49 struct CustomRegistryInterface : public TensorFlowRegistryInterfaceBase { in TEST()50 using TensorFlowRegistryInterfaceBase::TensorFlowRegistryInterfaceBase; in TEST()
28 class TensorFlowOpRegistryInterface : public TensorFlowRegistryInterfaceBase {33 : TensorFlowRegistryInterfaceBase(dialect), registry_(registry) {} in TensorFlowOpRegistryInterface()
157 getRegisteredInterface<TensorFlowRegistryInterfaceBase>()) { in getRegisteredInterfaceForOp()159 return static_cast<TensorFlowRegistryInterfaceBase::Concept *>( in getRegisteredInterfaceForOp()160 const_cast<TensorFlowRegistryInterfaceBase *>(instance)); in getRegisteredInterfaceForOp()