Searched refs:xla_func_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 358 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveHeadOutsideCompilationToHost() argument
749 << DumpGraphToFile(absl::StrCat("move_head_oc_host_", xla_func_name), in MoveHeadOutsideCompilationToHost()
752 << DumpGraphToFile(absl::StrCat("move_head_oc_xla_", xla_func_name), in MoveHeadOutsideCompilationToHost()
760 Status RemoveUnusedXlaInput(const string& xla_func_name, Graph* g, in RemoveUnusedXlaInput() argument
984 absl::StrCat("remove_unused_input_host_", xla_func_name), *g); in RemoveUnusedXlaInput()
987 absl::StrCat("remove_unused_input_xla_", xla_func_name), in RemoveUnusedXlaInput()
999 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveTailOutsideCompilationToHost() argument
1231 << DumpGraphToFile(absl::StrCat("move_tail_oc_host_", xla_func_name), in MoveTailOutsideCompilationToHost()
1234 << DumpGraphToFile(absl::StrCat("move_tail_oc_xla_", xla_func_name), in MoveTailOutsideCompilationToHost()
1241 const string& outside_compilation_attr_name, const string& xla_func_name, in ReplaceArgUsedByOutsideCompilationWithPlaceholder() argument
[all …]
|