Searched refs:ret_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 533 NodeDefBuilder ret_builder(absl::StrCat("ret_", arg_idx), "_Retval"); in AddMatchingRetvalNode() local
534 ret_builder.Attr("T", data_type); in AddMatchingRetvalNode()
535 ret_builder.Attr("index", arg_idx); in AddMatchingRetvalNode()
536 ret_builder.Input(arg_node->name(), 0, data_type); in AddMatchingRetvalNode()
538 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in AddMatchingRetvalNode()
1600 NodeDefBuilder ret_builder( in RewriteHostWhileLoopCond() local
1602 ret_builder.Attr("T", DT_BOOL); in RewriteHostWhileLoopCond()
1603 ret_builder.Attr("index", 0); in RewriteHostWhileLoopCond()
1604 ret_builder.Input(recv_pred_node->name(), 0, DT_BOOL); in RewriteHostWhileLoopCond()
1606 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in RewriteHostWhileLoopCond()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1185 NodeDefBuilder ret_builder(absl::StrCat("ret_", i), in MoveTailOutsideCompilationToHost() local
1187 ret_builder.Attr("T", new_ret_types[i - old_num_replicated_outputs]); in MoveTailOutsideCompilationToHost()
1188 ret_builder.Attr("index", i); in MoveTailOutsideCompilationToHost()
1192 ret_builder.Input(src->name(), src_output, src->output_type(src_output)); in MoveTailOutsideCompilationToHost()
1194 TF_RETURN_IF_ERROR(ret_builder.Finalize(&ret_def)); in MoveTailOutsideCompilationToHost()
|