Searched refs:id_builder (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 722 NodeDefBuilder id_builder(absl::StrCat("lifted_arg_input_", index), in MoveHeadOutsideCompilationToHost() local
726 id_builder.Attr("T", std::vector<DataType>(num_replicas, input_dtype)); in MoveHeadOutsideCompilationToHost()
731 id_builder.Attr(kXlaOutsideCompilationInputsAttrName, in MoveHeadOutsideCompilationToHost()
733 id_builder.Input(inputs); in MoveHeadOutsideCompilationToHost()
735 TF_RETURN_IF_ERROR(id_builder.Finalize(&id_def)); in MoveHeadOutsideCompilationToHost()
1290 NodeDefBuilder id_builder(absl::StrCat(oc_identifier, "_inputs"), in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1293 id_builder.Attr("T", dtypes); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1294 id_builder.Attr(kXlaOutsideCompilationInputsAttrName, oc_identifier); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1312 id_builder.Input(inputs); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1314 TF_RETURN_IF_ERROR(id_builder.Finalize(&id_def)); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 325 NodeDefBuilder id_builder(opts.new_name(src->name()), "Identity", in AddRecv() local
327 id_builder.Device(dst->assigned_device_name()) in AddRecv()
330 *status = id_builder.Finalize(id, /*consume=*/true); in AddRecv()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 962 NodeBuilder id_builder(replacee->name(), "Identity"); in AddIdentityNode() local
963 id_builder.Input(if_node, port); in AddIdentityNode()
968 id_builder.Attr(kXlaOutsideCompilationAttrName, outside_compilation); in AddIdentityNode()
971 TF_RETURN_IF_ERROR(id_builder.Finalize(graph_, &id)); in AddIdentityNode()
|