Home
last modified time | relevance | path

Searched refs:op_registry (Results 1 – 25 of 64) 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.cc1172 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1212 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1275 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1340 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1372 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1393 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1415 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1440 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1516 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F()
1553 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.cc684 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in ModifyNodeDefForImport()
899 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def.op(), &op_def)); in IsNodeFullyMapped()
1171 TF_RETURN_IF_ERROR(g_->op_registry()->LookUpOpDef(node_def->op(), &op_def)); in PopulateMissingUnusedInputMapKeys()
1209 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph()
1218 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph()
1262 ShapeRefiner default_refiner(gdef.versions().producer(), g->op_registry()); in ImportGraphDef()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc162 cond_call_builder_(NewName("cond"), cond_fn_name, graph->op_registry(), in LowerWhileHelper()
164 body_call_builder_(NewName("body"), body_fn_name, graph->op_registry(), in LowerWhileHelper()
199 graph_->op_registry(), &debug_info_) in CreateEnterNodes()
218 graph_->op_registry(), &debug_info_) in CreateEnterNodes()
233 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(), in CreateMergeNodes()
254 graph_->op_registry(), &debug_info_) in CreateCondFuncCallNode()
275 graph_->op_registry(), &debug_info_) in CreateSwitchNodes()
304 graph_->op_registry(), &debug_info_) in CreateBodyFuncCallNode()
318 graph_->op_registry(), &debug_info_) in CreateExitNodes()
341 graph_->op_registry(), &debug_info_) in CreateNextIterationNodes()
Dlower_if_op.cc110 then_call_builder_(NewName("then"), then_fn_name, graph->op_registry(), in CondBuilder()
112 else_call_builder_(NewName("else"), else_fn_name, graph->op_registry(), in CondBuilder()
124 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
131 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
136 graph_->op_registry(), &debug_info_) in CreatePivotNodes()
160 graph_->op_registry(), &debug_info) in AddInput()
201 NodeBuilder(graph_->NewName("output"), "Merge", graph_->op_registry(), in AddOutputs()
217 graph_->op_registry(), &debug_info_) in AddOutputs()
Dmemory_types.cc66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
209 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
Dfunction_testlib.cc118 NodeDefBuilder builder(op_name, fn_name, scope->graph()->op_registry()); in Call()
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc55 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument
57 return AddDefaultAttrsToGraphDef(graph_def, op_registry, node_offset, false); in AddDefaultAttrsToGraphDef()
61 const OpRegistryInterface& op_registry, in AddDefaultAttrsToGraphDef() argument
73 Status s = op_registry.LookUpOpDef(node_def->op(), &op_def); in AddDefaultAttrsToGraphDef()
212 const OpRegistryInterface& op_registry, in StrippedOpListForGraph() argument
221 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def)); in StrippedOpListForGraph()
Dgraph_def_util.h56 const OpRegistryInterface& op_registry,
62 const OpRegistryInterface& op_registry,
116 const OpRegistryInterface& op_registry,
Dmemory_types.cc75 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() argument
81 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def)); in MemoryTypesForNode()
Dmemory_types.h31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
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/ops/
Dshape_function_test.cc25 OpRegistry* op_registry = OpRegistry::Global(); in TEST() local
27 op_registry->GetOpRegistrationData(&op_data); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc82 const OpRegistryInterface& op_registry) { in IsOutputPortRefValue() argument
84 Status s = op_registry.LookUp(node.op(), &op_reg_data); in IsOutputPortRefValue()
97 const OpRegistryInterface& op_registry) { in CanRemoveNode() argument
112 } else if (IsOutputPortRefValue(*input.node, input.port_id, op_registry)) { in CanRemoveNode()
455 OpRegistryInterface* op_registry = OpRegistry::Global(); in Optimize() local
483 if (CanRemoveNode(node, graph_view, function_names, *op_registry)) { in Optimize()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc58 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
Dpartially_decluster_pass.cc61 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
82 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops_test.cc330 TF_ASSERT_OK(NodeBuilder("range_decode", "RangeDecode", g.op_registry()) in TEST_F()
337 ShapeRefiner refiner{g.versions().producer(), g.op_registry()}; in TEST_F()
431 TF_CHECK_OK(NodeBuilder("range_encode", "RangeEncode", g->op_registry()) in CreateRangeEncodeFullBroadcastGraph()
480 TF_CHECK_OK(NodeBuilder("range_decode", "RangeDecode", g->op_registry()) in CreateRangeDecodeFullBroadcastGraph()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc604 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry) { in IsStateful() argument
607 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
621 const OpRegistryInterface* op_registry) { in IsFreeOfSideEffect() argument
628 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()
Dop_types.h202 bool IsStateful(const NodeDef node, const OpRegistryInterface* op_registry);
206 const OpRegistryInterface* op_registry);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc179 OpRegistryInterface* op_registry = OpRegistry::Global(); in RegisterCompilationKernels() local
211 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def); in RegisterCompilationKernels()
217 dynamic_cast<OpRegistry*>(op_registry)->DebugString(true)); in RegisterCompilationKernels()

123