| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_util.cc | 44 NodeDef ret_def; in BuildRetvalNode() local
45 ret_def.set_op(kRetValOp); in BuildRetvalNode()
46 ret_def.set_name(absl::StrCat(kRetValOp, index)); in BuildRetvalNode()
47 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
48 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
49 return AddNodeDefToGraph(ret_def, graph); in BuildRetvalNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 338 const OpDef::ArgDef& ret_def, AttrSlice attrs, in AddReturnNode() argument
341 auto ret_iter = ret_map.find(ret_def.name()); in AddReturnNode()
343 return errors::InvalidArgument("Return ", ret_def.name(), " missing."); in AddReturnNode()
347 TF_RETURN_IF_ERROR(ArgNumType(attrs, ret_def, &is_type_list, &dtypes)); in AddReturnNode()
351 return errors::InvalidArgument("Return ", ret_def.name(), " -> ", in AddReturnNode()
355 return errors::InvalidArgument("Invalid ret types ", ret_def.name(), in AddReturnNode()
361 string name = strings::StrCat(ret_def.name(), "_RetVal"); in AddReturnNode()
372 DataType dtype = ret_def.is_ref() ? MakeRefType(dtypes[i]) : dtypes[i]; in AddReturnNode()
798 for (const OpDef::ArgDef& ret_def : sig.output_arg()) { in InstantiateFunction() local
799 s = helper.AddReturnNode(ret_def, attr_values, fdef.ret(), ints_on_device, in InstantiateFunction()
[all …]
|
| D | function.h | 149 gtl::ArraySlice<std::pair<string, string>> ret_def,
161 gtl::ArraySlice<std::pair<string, string>> ret_def);
166 gtl::ArraySlice<string> ret_def,
172 gtl::ArraySlice<string> ret_def,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 150 for (const OpDef::ArgDef& ret_def : fdef->signature().output_arg()) { in FillOutputDevices() local
152 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 | 537 NodeDef ret_def; in AddMatchingRetvalNode() local
538 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in AddMatchingRetvalNode()
540 Node* ret_node = function_body.graph->AddNode(ret_def, &s); in AddMatchingRetvalNode()
1605 NodeDef ret_def; in RewriteHostWhileLoopCond() local
1606 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in RewriteHostWhileLoopCond()
1607 Node* ret_node = cond_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopCond()
1664 NodeDef ret_def; in RewriteHostWhileLoopBody() local
1665 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in RewriteHostWhileLoopBody()
1667 Node* ret_node = body_graph->AddNode(ret_def, &s); in RewriteHostWhileLoopBody()
|
| D | encapsulate_subgraphs_pass.cc | 524 NodeDef ret_def; in RecordResult() local
532 Status s = builder.Finalize(&ret_def); in RecordResult()
534 Node* ret = graph_->AddNode(ret_def, &s); in RecordResult()
|
| D | extract_outside_compilation_pass_test.cc | 755 NodeDef ret_def; in TEST_F() local
760 .Finalize(&ret_def)); in TEST_F()
761 Node *ret_node = g->AddNode(ret_def, &s); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 1008 for ret_def in op_def.output_arg:
1009 if ret_def.number_attr or ret_def.type_list_attr:
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1193 NodeDef ret_def; in MoveTailOutsideCompilationToHost() local
1194 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in MoveTailOutsideCompilationToHost()
1196 Node* ret_node = xla_graph->AddNode(ret_def, &s); in MoveTailOutsideCompilationToHost()
|