Searched refs:ret_def (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_control_flow_util.cc | 43 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/ |
D | function.cc | 312 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 …]
|
D | function.h | 136 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/ |
D | partitioned_function_ops.cc | 139 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/ |
D | extract_outside_compilation_pass.cc | 1081 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()
|
D | extract_outside_compilation_pass_test.cc | 873 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()
|
D | encapsulate_subgraphs_pass.cc | 817 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()
|