Searched refs:ph_builder (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 697 NodeDefBuilder ph_builder(absl::StrCat("placeholder_", n->name()), in MoveHeadOutsideCompilationToHost() local
701 ph_builder.Attr("dtype", dtype); in MoveHeadOutsideCompilationToHost()
702 ph_builder.Attr(kXlaIsLiftedArgAttrName, true); in MoveHeadOutsideCompilationToHost()
703 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in MoveHeadOutsideCompilationToHost()
705 TF_RETURN_IF_ERROR(ph_builder.Finalize(&ph_def)); in MoveHeadOutsideCompilationToHost()
1333 NodeDefBuilder ph_builder(xla_graph->NewName("ph_for_arg_in_oc_"), in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1335 ph_builder.Attr("dtype", dtype); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1340 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1341 ph_builder.Attr(kXlaOutsideCompilationInputsAttrName, oc_identifier); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1342 ph_builder.Attr(kXlaIsPlaceholderForArg, true); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
[all …]
|