Home
last modified time | relevance | path

Searched refs:OpRegistryInterface (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.h56 const OpRegistryInterface& op_registry,
62 const OpRegistryInterface& op_registry,
96 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry,
97 const OpRegistryInterface& producer_op_registry,
116 const OpRegistryInterface& op_registry,
Dgraph_def_util.cc55 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
61 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
85 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
86 const OpRegistryInterface& producer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
137 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromGraphDef()
138 const OpRegistryInterface& producer_op_registry, in RemoveNewDefaultAttrsFromGraphDef()
212 const OpRegistryInterface& op_registry, in StrippedOpListForGraph()
Dop.h41 class OpRegistryInterface {
43 virtual ~OpRegistryInterface();
65 class OpRegistry : public OpRegistryInterface {
167 class OpListOpRegistry : public OpRegistryInterface {
Dop.cc35 OpRegistryInterface::~OpRegistryInterface() {} in ~OpRegistryInterface()
37 Status OpRegistryInterface::LookUpOpDef(const string& op_type_name, in LookUpOpDef()
Dmemory_types.h31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
Dfunction.h312 class FunctionLibraryDefinition : public OpRegistryInterface {
316 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
433 const OpRegistryInterface* default_registry() const { in default_registry()
465 const OpRegistryInterface* const default_registry_;
Dmemory_types.cc75 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode()
Dnode_def_builder.h67 const OpRegistryInterface* op_registry = OpRegistry::Global(),
Dnode_def_builder.cc41 const OpRegistryInterface* op_registry, in NodeDefBuilder()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.h26 class OpRegistryInterface; variable
58 const OpRegistryInterface& ops, int32 graph_def_version, bool* evaluated,
Dshape_refiner.h91 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
94 ShapeRefiner(const VersionDef& versions, const OpRegistryInterface* ops);
264 const OpRegistryInterface* const ops_registry_;
Dshape_refiner.cc43 const OpRegistryInterface* ops) in ShapeRefiner()
49 const OpRegistryInterface* ops) in ShapeRefiner()
Deval_const_tensor.cc309 const OpRegistryInterface& ops, in EvaluateConstantTensor()
/external/tensorflow/tensorflow/core/graph/
Dvalidate.h35 const OpRegistryInterface& op_registry);
41 const GraphDef& graph_def, const OpRegistryInterface& op_registry);
Dgraph_def_builder.h120 const OpRegistryInterface* op_registry() const { in op_registry()
146 const OpRegistryInterface* op_registry = OpRegistry::Global())
154 const OpRegistryInterface* op_registry = OpRegistry::Global())
Dvalidate.cc32 const OpRegistryInterface& op_registry) { in ValidateGraphDef()
47 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry()
Dnode_builder.h80 const OpRegistryInterface* op_registry = OpRegistry::Global(),
Dnode_builder.cc41 const OpRegistryInterface* op_registry, in NodeBuilder()
Dgraph.h456 explicit Graph(const OpRegistryInterface* ops);
607 const OpRegistryInterface* op_registry() const { return &ops_; } in op_registry()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_env.h34 class OpRegistryInterface; variable
69 const OpRegistryInterface* ops = nullptr;
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc82 const OpRegistryInterface& op_registry) { in IsOutputPortRefValue()
97 const OpRegistryInterface& op_registry) { in CanRemoveNode()
455 OpRegistryInterface* op_registry = OpRegistry::Global(); in Optimize()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h202 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry);
206 const OpRegistryInterface* op_registry);
Dop_types.cc604 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry) { in IsStateful()
621 const OpRegistryInterface* op_registry) { in IsFreeOfSideEffect()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.h41 const tf2xla::Config& config, const OpRegistryInterface* op_registry,
Dxla_op_registry.cc179 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels()

12