Home
last modified time | relevance | path

Searched defs:op_registry (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.h120 const OpRegistryInterface* op_registry() const { in op_registry() function
147 : graph_(op_registry), opts_(&graph_, &status_) {} in graph_() argument
155 : graph_(op_registry), opts_(&graph_, nullptr) {} in graph_() argument
Dvalidate.cc32 const OpRegistryInterface& op_registry) { in ValidateGraphDef()
47 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry()
59 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation()
Dnode_builder.cc41 const OpRegistryInterface* op_registry, in NodeBuilder()
Dgraph.h607 const OpRegistryInterface* op_registry() const { return &ops_; } in op_registry() function
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc55 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
61 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
212 const OpRegistryInterface& op_registry, in StrippedOpListForGraph()
Dmemory_types.cc75 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode()
Dnode_def_builder.cc41 const OpRegistryInterface* op_registry, in NodeDefBuilder()
Dop_kernel.cc1348 Status ValidateKernelRegistrations(const OpRegistryInterface& op_registry) { in ValidateKernelRegistrations()
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc25 OpRegistry* op_registry = OpRegistry::Global(); in TEST() local
/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() local
Dloop_optimizer.cc206 OpRegistryInterface* op_registry = OpRegistry::Global(); in HandleInvariantNode() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc83 OpRegistry* op_registry = OpRegistry::Global(); in ExtractExtraProperties() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc179 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local
Dtf2xla_util.cc333 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds()
/external/tensorflow/tensorflow/core/grappler/
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/tests/
Drandomized_tests.cc209 OpRegistryInterface* op_registry = OpRegistry::Global(); in BuildGraph() local