Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/graph/
Dvalidate.cc32 const OpRegistryInterface& op_registry) { in ValidateGraphDef() argument
38 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def)); in ValidateGraphDef()
47 const GraphDef& graph_def, const OpRegistryInterface& op_registry) { in ValidateGraphDefAgainstOpRegistry() argument
49 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(&copy, op_registry, 0)); in ValidateGraphDefAgainstOpRegistry()
50 return ValidateGraphDef(copy, op_registry); in ValidateGraphDefAgainstOpRegistry()
59 void GetOpListForValidation(OpList* op_list, const OpRegistry& op_registry) { in GetOpListForValidation() argument
60 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.h35 const OpRegistryInterface& op_registry);
41 const GraphDef& graph_def, const OpRegistryInterface& op_registry);
52 OpList* op_list, const OpRegistry& op_registry = *OpRegistry::Global());
Dgraph_def_builder.cc104 opts.op_registry()); in SourceOp()
112 opts.op_registry()); in UnaryOp()
121 opts.op_registry()); in BinaryOp()
Dgraph_constructor_test.cc1171 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1211 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1274 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1339 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1371 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1392 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1414 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1439 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1515 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1552 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
[all …]
Dtestlib.cc140 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce()
185 TF_CHECK_OK(NodeBuilder(g->NewName("n"), op, g->op_registry()) in RandomNumberGenerator()
227 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Unary()
235 TF_CHECK_OK(NodeBuilder(g->NewName("n"), func, g->op_registry()) in Binary()
244 auto b = NodeBuilder(g->NewName("n"), func, g->op_registry()); in Multi()
266 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Roll", g->op_registry()) in Roll()
Dnode_builder.cc41 const OpRegistryInterface* op_registry, in NodeBuilder() argument
43 : def_builder_(name, op_name, op_registry, debug) {} in NodeBuilder()
Dgraph_constructor.cc847 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport()
1061 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped()
1247 g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in Convert()
1417 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in PopulateMissingUnusedInputMapKeys()
1455 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1464 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1473 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph()
1518 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.h58 const OpRegistryInterface& op_registry,
64 const OpRegistryInterface& op_registry,
111 void StripDefaultAttributes(const OpRegistryInterface& op_registry,
130 const OpRegistryInterface& op_registry,
Dgraph_def_util.cc56 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument
58 return AddDefaultAttrsToGraphDef(graph_def, op_registry, node_offset, false); in AddDefaultAttrsToGraphDef()
62 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument
74 Status s = op_registry.LookUpOpDef(node_def->op(), &op_def); in AddDefaultAttrsToGraphDef()
167 void StripDefaultAttributes(const OpRegistryInterface& op_registry, in StripDefaultAttributes() argument
174 Status s = op_registry.LookUp(node->op(), &op_reg_data); in StripDefaultAttributes()
249 const OpRegistryInterface& op_registry, in StrippedOpListForGraph() argument
258 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
Dmemory_types.h31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
Dmemory_types.cc81 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() argument
87 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
Dnode_def_builder.cc41 const OpRegistryInterface* op_registry, in NodeDefBuilder() argument
44 const Status status = op_registry->LookUpOpDef(string(op_name), &op_def_); in NodeDefBuilder()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc113 then_call_builder_(NewName("then"), then_fn.name(), graph->op_registry(), in CondBuilder()
115 else_call_builder_(NewName("else"), else_fn.name(), graph->op_registry(), in CondBuilder()
135 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
142 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
147 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
171 graph_->op_registry(), &debug_info) in AddInput()
212 NodeBuilder(NewName("output"), "Merge", graph_->op_registry(), in AddOutputs()
231 graph_->op_registry(), &debug_info_) in AddOutputs()
Dlower_while_op.cc189 cond_call_builder_(NewName("cond"), cond_fn.name(), graph->op_registry(), in LowerWhileHelper()
191 body_call_builder_(NewName("body"), body_fn.name(), graph->op_registry(), in LowerWhileHelper()
244 graph_->op_registry(), &debug_info_) in CreateEnterNodes()
266 graph_->op_registry(), &debug_info_) in CreateEnterNodes()
284 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(), in CreateMergeNodes()
310 graph_->op_registry(), &debug_info_) in CreateCondFuncCallNode()
336 NodeBuilder(NewName(op_name), op_type, graph_->op_registry(), in CreateSwitchNodes()
372 graph_->op_registry(), &debug_info_) in CreateBodyFuncCallNode()
393 NodeBuilder(NewName("exit"), "Exit", graph_->op_registry(), in CreateExitNodes()
446 graph_->op_registry(), &debug_info_) in CreateNextIterationNodes()
Dlower_case_op.cc111 branch_fn_names[b], graph->op_registry(), in CaseBuilder()
124 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
134 "Identity", graph_->op_registry(), in CreatePivotNodes()
160 graph_->op_registry(), &debug_info) in AddInput()
211 graph_->op_registry(), &debug_info_) in AddOutputs()
233 graph_->op_registry(), &debug_info_) in AddOutputs()
Dmemory_types.cc67 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
216 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc25 OpRegistry* op_registry = OpRegistry::Global(); in TEST() local
27 op_registry->GetOpRegistrationData(&op_data); in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util_test.cc89 auto op_reg = opts.op_registry(); in TEST()
124 auto op_reg = opts.op_registry(); in TEST()
164 auto op_reg = opts.op_registry(); in TEST()
207 auto op_reg = opts.op_registry(); in TEST()
256 auto op_reg = opts.op_registry(); in TEST()
319 auto op_reg = opts.op_registry(); in TEST()
395 auto op_reg = opts.op_registry(); in TEST()
452 auto op_reg = opts.op_registry(); in TEST()
502 auto op_reg = opts.op_registry(); in TEST()
Dgraph_compiler_util.cc285 TF_RETURN_IF_ERROR(AddPlaceholdersForFeeds(config, g->op_registry(), in InitGraph()
295 &second_copy_def, *g->op_registry(), /*node_offset=*/0)); in InitGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc90 const OpRegistryInterface& op_registry) { in IsOutputPortRefValue() argument
92 Status s = op_registry.LookUp(node.op(), &op_reg_data); in IsOutputPortRefValue()
105 const OpRegistryInterface& op_registry) { in CanRemoveNode() argument
127 } else if (IsOutputPortRefValue(*input.node, input.port_id, op_registry)) { in CanRemoveNode()
452 OpRegistryInterface* op_registry = OpRegistry::Global(); in Optimize() local
480 if (CanRemoveNode(node, graph_view, function_names, *op_registry)) { in Optimize()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc69 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
Dpartially_decluster_pass.cc73 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
94 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h215 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry);
219 const OpRegistryInterface* op_registry);
Dop_types.cc644 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry) { in IsStateful() argument
647 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
661 const OpRegistryInterface* op_registry) { in IsFreeOfSideEffect() argument
668 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()

123