Home
last modified time | relevance | path

Searched refs:op_registry (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dvalidate.cc30 const OpRegistryInterface& op_registry) { in ValidateGraphDef() argument
36 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def)); in ValidateGraphDef()
45 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry() argument
47 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(&copy, op_registry, 0)); in ValidateGraphDefAgainstOpRegistry()
48 return ValidateGraphDef(copy, op_registry); in ValidateGraphDefAgainstOpRegistry()
57 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation() argument
58 op_registry.Export(false, op_list); in GetOpListForValidation()
Dgraph_def_builder.h120 const OpRegistryInterface* op_registry() const { in op_registry() function
121 return graph_->op_registry(); in op_registry()
146 const OpRegistryInterface* op_registry = OpRegistry::Global())
147 : graph_(op_registry), opts_(&graph_, &status_) {} in graph_() argument
154 const OpRegistryInterface* op_registry = OpRegistry::Global())
155 : graph_(op_registry), opts_(&graph_, nullptr) {} in graph_() argument
Dvalidate.h34 const OpRegistryInterface& op_registry);
40 const GraphDef& graph_def, const OpRegistryInterface& op_registry);
51 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
Dgraph_constructor_test.cc1167 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1207 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1270 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1335 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1367 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1388 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1410 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1435 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1507 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1544 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
[all …]
Dgraph_def_builder.cc104 opts.op_registry()); in SourceOp()
112 opts.op_registry()); in UnaryOp()
121 opts.op_registry()); in BinaryOp()
Dtestlib.cc152 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce()
197 TF_CHECK_OK(NodeBuilder(g->NewName("n"), op, g->op_registry()) in RandomNumberGenerator()
239 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Unary()
247 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Binary()
256 auto b = NodeBuilder(g->NewName("n"), func, g->op_registry()); in Multi()
278 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Roll", g->op_registry()) in Roll()
Dnode_builder.cc39 const OpRegistryInterface* op_registry) in NodeBuilder() argument
40 : def_builder_(name, op_name, op_registry) {} in NodeBuilder()
Dgraph_constructor.cc647 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport()
853 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped()
1135 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in PopulateMissingUnusedInputMapKeys()
1171 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1180 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph()
1224 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
Dnode_builder.h79 const OpRegistryInterface* op_registry = OpRegistry::Global());
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc54 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument
66 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def->op(), &op_def)); in AddDefaultAttrsToGraphDef()
201 const OpRegistryInterface& op_registry, in StrippedOpListForGraph() argument
210 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
Dgraph_def_util.h56 const OpRegistryInterface& op_registry,
110 const OpRegistryInterface& op_registry,
Dmemory_types.cc70 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() argument
76 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
Dmemory_types.h31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
Ddataset.cc105 opts->op_registry()); in AddDataset()
191 Status s = b_->opts().op_registry()->LookUpOpDef(op_type_name, &op_def); in HasAttr()
Dnode_def_builder.cc40 const OpRegistryInterface* op_registry) { in NodeDefBuilder() argument
42 const Status status = op_registry->LookUpOpDef(op_name.ToString(), &op_def_); in NodeDefBuilder()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc98 const tf2xla::Config& config, const OpRegistryInterface* op_registry, in AddPlaceholdersForFeeds() argument
146 AddDefaultAttrsToGraphDef(&gd, *op_registry, 0 /*node_offset*/)); in AddPlaceholdersForFeeds()
149 Graph g(op_registry); in AddPlaceholdersForFeeds()
Dxla_op_registry.cc159 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local
164 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def); in RegisterCompilationKernels()
169 dynamic_cast<OpRegistry*>(op_registry)->DebugString(true)); in RegisterCompilationKernels()
Dtf2xla_util.h38 const tf2xla::Config& config, const OpRegistryInterface* op_registry,
Dtf2xla.cc328 TF_RETURN_IF_ERROR(AddPlaceholdersForFeeds(config, g->op_registry(), in InitGraph()
338 &second_copy_def, *g->op_registry(), /*node_offset=*/0)); in InitGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_rewriter.cc32 OpRegistryInterface* op_registry = OpRegistry::Global(); in GraphRewriter() local
38 Status s = op_registry->LookUp(node.op(), &op_reg_data); in GraphRewriter()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc55 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
177 builder.opts().op_registry()); in TEST()
312 builder.opts().op_registry()); in TEST()
325 builder.opts().op_registry()); in TEST()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops_test.cc331 TF_ASSERT_OK(NodeBuilder("range_decode", "RangeDecode", g.op_registry()) in TEST_F()
338 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F()
432 TF_CHECK_OK(NodeBuilder("range_encode", "RangeEncode", g->op_registry()) in CreateRangeEncodeFullBroadcastGraph()
481 TF_CHECK_OK(NodeBuilder("range_decode", "RangeDecode", g->op_registry()) in CreateRangeDecodeFullBroadcastGraph()
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
208 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
Dfunction_testlib.cc62 NodeDefBuilder builder(op_name, fn_name, scope->graph()->op_registry()); in Call()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc86 OpRegistry* op_registry = OpRegistry::Global(); in ExtractExtraProperties() local
88 auto s = op_registry->LookUpOpDef(node.op(), &op_def); in ExtractExtraProperties()

12