Home
last modified time | relevance | path

Searched refs:xla_cluster_attr_name (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.h47 const string& xla_cluster_attr_name, in RewriteOutsideCompilationSubgraphFn() argument
50 : xla_cluster_attr_name_(xla_cluster_attr_name), in RewriteOutsideCompilationSubgraphFn()
90 const string& xla_cluster_attr_name,
103 const string& xla_cluster_attr_name,
Dextract_outside_compilation_pass.cc1392 const string& xla_cluster_attr_name, in BuildHostGraphForIfNode() argument
1414 recv_pred_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForIfNode()
1446 if_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForIfNode()
1552 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopCond() argument
1589 recv_pred_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in RewriteHostWhileLoopCond()
1628 const string& host_transfer_key, const string& xla_cluster_attr_name, in RewriteHostWhileLoopBody() argument
1687 const string& xla_cluster_attr_name, in BuildHostGraphForWhileNode() argument
1703 xla_cluster_attr_name, xla_cluster_name, outside_compilation_attr_name, in BuildHostGraphForWhileNode()
1709 xla_cluster_attr_name, xla_cluster_name, outside_compilation_attr_name, in BuildHostGraphForWhileNode()
1725 while_builder.Attr(xla_cluster_attr_name, xla_cluster_name); in BuildHostGraphForWhileNode()
[all …]
Dextract_outside_compilation_pass_test.cc239 const string &xla_cluster_attr_name, in ExtractOutsideCompilationTest() argument
254 xla_cluster_attr_name, outside_compilation_attr_name, xla_cluster_name, in ExtractOutsideCompilationTest()