Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.h58 const OpRegistryInterface& op_registry,
64 const OpRegistryInterface& op_registry,
98 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry,
99 const OpRegistryInterface& producer_op_registry,
111 void StripDefaultAttributes(const OpRegistryInterface& op_registry,
130 const OpRegistryInterface& op_registry,
Dgraph_def_util.cc56 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
62 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
86 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
87 const OpRegistryInterface& producer_op_registry, in RemoveNewDefaultAttrsFromNodeDef()
138 GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry, in RemoveNewDefaultAttrsFromGraphDef()
139 const OpRegistryInterface& producer_op_registry, in RemoveNewDefaultAttrsFromGraphDef()
167 void StripDefaultAttributes(const OpRegistryInterface& op_registry, in StripDefaultAttributes()
249 const OpRegistryInterface& op_registry, in StrippedOpListForGraph()
Dop.h41 class OpRegistryInterface {
43 virtual ~OpRegistryInterface();
66 class OpRegistry : public OpRegistryInterface {
101 std::function<Status(const OpRegistryInterface&)> validator) { in RegisterValidator()
170 std::function<Status(const OpRegistryInterface&)> op_registry_validator_;
178 class OpListOpRegistry : public OpRegistryInterface {
Dnode_properties.h26 class OpRegistryInterface; variable
50 const OpRegistryInterface* op_registry,
Dop.cc34 Status DefaultValidator(const OpRegistryInterface& op_registry) { in DefaultValidator()
41 OpRegistryInterface::~OpRegistryInterface() {} in ~OpRegistryInterface()
43 Status OpRegistryInterface::LookUpOpDef(const string& op_type_name, in LookUpOpDef()
Dmemory_types.h31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
Dnode_properties.cc25 NodeDef node_def, const OpRegistryInterface* op_registry, in CreateFromNodeDef()
Dfunction.h366 class FunctionLibraryDefinition : public OpRegistryInterface {
384 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
501 const OpRegistryInterface* default_registry() const { in default_registry()
584 const OpRegistryInterface* const default_registry_;
Dnode_properties_test.cc31 class MockOpRegistry : public OpRegistryInterface {
Dmemory_types.cc82 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode()
Dnode_def_builder.h69 const OpRegistryInterface* op_registry = OpRegistry::Global(),
/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.h505 explicit Graph(const OpRegistryInterface* ops);
658 const OpRegistryInterface* op_registry() const { return &ops_; } in op_registry()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.h26 class OpRegistryInterface; variable
61 const OpRegistryInterface& ops, int32 graph_def_version, bool* evaluated,
Dshape_refiner.h75 ShapeRefiner(int graph_def_version, const OpRegistryInterface* ops);
78 ShapeRefiner(const VersionDef& versions, const OpRegistryInterface* ops);
294 const OpRegistryInterface* const ops_registry_;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_env.h34 class OpRegistryInterface; variable
69 const OpRegistryInterface* ops = nullptr;
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.h32 GraphDef& gdef, const OpRegistryInterface* default_registry);
Dupgrade_graph.cc44 GraphDef& gdef, const OpRegistryInterface* default_registry) { in GenerateResourceSharedNameIfEmpty()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc91 const OpRegistryInterface& op_registry) { in IsOutputPortRefValue()
106 const OpRegistryInterface& op_registry) { in CanRemoveNode()
458 OpRegistryInterface* op_registry = OpRegistry::Global(); in Optimize()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h227 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry);
231 const OpRegistryInterface* op_registry);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.h42 const tf2xla::Config& config, const OpRegistryInterface* op_registry,

12