Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dvalidate.cc30 const OpRegistryInterface& op_registry) { in ValidateGraphDef()
45 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry()
57 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation()
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
Dnode_builder.cc39 const OpRegistryInterface* op_registry) in NodeBuilder()
Dgraph.h542 const OpRegistryInterface* op_registry() const { return &ops_; } in op_registry() function
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc54 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef()
201 const OpRegistryInterface& op_registry, in StrippedOpListForGraph()
Dmemory_types.cc70 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode()
Dnode_def_builder.cc40 const OpRegistryInterface* op_registry) { in NodeDefBuilder()
Dop_kernel.cc1190 Status ValidateKernelRegistrations(const OpRegistryInterface& op_registry) { in ValidateKernelRegistrations()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_rewriter.cc32 OpRegistryInterface* op_registry = OpRegistry::Global(); in GraphRewriter() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc98 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds()
Dxla_op_registry.cc159 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc86 OpRegistry* op_registry = OpRegistry::Global(); in ExtractExtraProperties() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc188 OpRegistryInterface* op_registry = OpRegistry::Global(); in BuildGraph() local