Home
last modified time | relevance | path

Searched defs:op_registry (Results 1 – 17 of 17) 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.h633 const OpRegistryInterface* op_registry() const { return &ops_; } in op_registry() function
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc56 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
62 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
167 void StripDefaultAttributes(const OpRegistryInterface& op_registry, in StripDefaultAttributes()
249 const OpRegistryInterface& op_registry, in StrippedOpListForGraph()
Dmemory_types.cc81 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode()
Dop.cc34 Status DefaultValidator(const OpRegistryInterface& op_registry) { in DefaultValidator()
Dnode_def_builder.cc41 const OpRegistryInterface* op_registry, in NodeDefBuilder()
Dop_kernel.cc1634 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.cc90 const OpRegistryInterface& op_registry) { in IsOutputPortRefValue()
105 const OpRegistryInterface& op_registry) { in CanRemoveNode()
452 OpRegistryInterface* op_registry = OpRegistry::Global(); in Optimize() local
Dloop_optimizer.cc204 OpRegistryInterface* op_registry = OpRegistry::Global(); in HandleInvariantNode() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc84 OpRegistry* op_registry = OpRegistry::Global(); in ExtractExtraProperties() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc183 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local
Dtf2xla_util.cc328 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc644 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry) { in IsStateful()
661 const OpRegistryInterface* op_registry) { in IsFreeOfSideEffect()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc209 OpRegistryInterface* op_registry = OpRegistry::Global(); in BuildGraph() local