Home
last modified time | relevance | path

Searched defs:xla_cluster_name (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.h49 const string& xla_cluster_name, const string& new_function_name) in RewriteOutsideCompilationSubgraphFn()
Dextract_outside_compilation_pass.cc55 StatusOr<Node*> AddHostComputeKeyPlaceholder(const string& xla_cluster_name, in AddHostComputeKeyPlaceholder()
990 const string& xla_cluster_name, const string& outside_compilation_attr_name, in ConstructHostGraph()
1358 Status ReplaceKeyPlaceholderWithArgNode(const string& xla_cluster_name, in ReplaceKeyPlaceholderWithArgNode()
1407 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForIfNode()
1564 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopCond()
1638 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopBody()
1695 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForWhileNode()
1761 const string& xla_cluster_attr_name, const string& xla_cluster_name, in BuildHostGraphForFuncCallNode()
1810 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForFuncCallNode()
1896 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForIfNode()
[all …]
Dextract_outside_compilation_pass_test.cc241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2934 string xla_cluster_name = n->name(); in Run() local