Searched defs:xla_cluster_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.h | 49 const string& xla_cluster_name, const string& new_function_name) in RewriteOutsideCompilationSubgraphFn()
|
| D | extract_outside_compilation_pass.cc | 55 const string& xla_cluster_name, Graph* g) { in AddHostComputeKeyPlaceholder()
982 const string& xla_cluster_name, const string& outside_compilation_attr_name, in ConstructHostGraph()
1344 Status ReplaceKeyPlaceholderWithArgNode(const string& xla_cluster_name, in ReplaceKeyPlaceholderWithArgNode()
1393 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForIfNode()
1553 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopCond()
1629 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopBody()
1688 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForWhileNode()
1756 const string& xla_cluster_attr_name, const string& xla_cluster_name, in BuildHostGraphForFuncCallNode()
1807 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForFuncCallNode()
1893 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForIfNode()
[all …]
|
| D | extract_outside_compilation_pass_test.cc | 241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2986 string xla_cluster_name = n->name(); in Run() local
|