Searched defs: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()
760 Status RemoveUnusedXlaInput(const string& xla_func_name, Graph* g, in RemoveUnusedXlaInput()
999 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveTailOutsideCompilationToHost()
1241 const string& outside_compilation_attr_name, const string& xla_func_name, in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1368 Status RemoveUnusedXlaOutput(const string& xla_func_name, Graph* g, in RemoveUnusedXlaOutput()
1479 Status RemoveEdgesBetweenArgAndRetval(const string& xla_func_name, Graph* g, in RemoveEdgesBetweenArgAndRetval()
2899 string xla_func_name = iter.second.func_name_attrs.name(); in ProcessHeadTailOutsideCompilation() local
|