Home
last modified time | relevance | path

Searched refs:ret_def (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc43 NodeDef ret_def; in BuildRetvalNode() local
44 ret_def.set_op(kRetValOp); in BuildRetvalNode()
45 ret_def.set_name(absl::StrCat(kRetValOp, index)); in BuildRetvalNode()
46 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
47 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
48 return AddNodeDefToGraph(ret_def, graph); in BuildRetvalNode()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc312 const OpDef::ArgDef& ret_def, AttrSlice attrs, in AddReturnNode() argument
315 auto ret_iter = ret_map.find(ret_def.name()); in AddReturnNode()
317 return errors::InvalidArgument("Return ", ret_def.name(), " missing."); in AddReturnNode()
321 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in AddReturnNode()
325 return errors::InvalidArgument("Return ", ret_def.name(), " -> ", in AddReturnNode()
329 return errors::InvalidArgument("Invalid ret types ", ret_def.name(), in AddReturnNode()
335 string name = strings::StrCat(ret_def.name(), "_RetVal"); in AddReturnNode()
751 for (const OpDef::ArgDef& ret_def : sig.output_arg()) { in InstantiateFunction() local
752 s = helper.AddReturnNode(ret_def, attr_values, fdef.ret(), ints_on_device, in InstantiateFunction()
755 errors::AppendToMessage(&s, "In function output ", Print(ret_def)); in InstantiateFunction()
[all …]
Dfunction.h136 gtl::ArraySlice<std::pair<string, string>> ret_def,
148 gtl::ArraySlice<std::pair<string, string>> ret_def);
153 gtl::ArraySlice<string> ret_def,
159 gtl::ArraySlice<string> ret_def,
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc139 for (const OpDef::ArgDef& ret_def : fdef->signature().output_arg()) { in FillOutputDevices() local
141 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in FillOutputDevices()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc1081 NodeDef ret_def; in RewriteHostWhileLoopCond() local
1082 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in RewriteHostWhileLoopCond()
1083 Node* ret_node = cond_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopCond()
1142 NodeDef ret_def; in RewriteHostWhileLoopBody() local
1143 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in RewriteHostWhileLoopBody()
1145 Node* ret_node = body_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopBody()
Dextract_outside_compilation_pass_test.cc873 NodeDef ret_def; in TEST_F() local
878 .Finalize(&ret_def)); in TEST_F()
879 Node *ret_node = g->AddNode(ret_def, &s); in TEST_F()
Dencapsulate_subgraphs_pass.cc817 NodeDef ret_def; in RecordResult() local
825 Status s = builder.Finalize(&ret_def); in RecordResult()
827 Node* ret = graph_->AddNode(ret_def, &s); in RecordResult()