Searched refs:xla_cluster_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 55 const string& xla_cluster_name, Graph* g) { in AddHostComputeKeyPlaceholder() argument
57 NodeDefBuilder builder(absl::StrCat(xla_cluster_name, "_key_placeholder"), in AddHostComputeKeyPlaceholder()
61 builder.Attr("_host_compute_call_node", xla_cluster_name); in AddHostComputeKeyPlaceholder()
982 const string& xla_cluster_name, const string& outside_compilation_attr_name, in ConstructHostGraph() argument
988 NodeDefBuilder sequencer_builder(absl::StrCat(xla_cluster_name, "_sequencer"), in ConstructHostGraph()
990 sequencer_builder.Attr("_xla_host_transfer_sequencer", xla_cluster_name); in ConstructHostGraph()
1000 AddHostComputeKeyPlaceholder(xla_cluster_name, host_graph->get())); in ConstructHostGraph()
1110 xla_cluster_name), in ConstructHostGraph()
1344 Status ReplaceKeyPlaceholderWithArgNode(const string& xla_cluster_name, in ReplaceKeyPlaceholderWithArgNode() argument
1369 AddHostComputeKeyPlaceholder(xla_cluster_name, g)); in ReplaceKeyPlaceholderWithArgNode()
[all …]
|
| D | extract_outside_compilation_pass.h | 49 const string& xla_cluster_name, const string& new_function_name) in RewriteOutsideCompilationSubgraphFn() argument
52 xla_cluster_name_(xla_cluster_name), in RewriteOutsideCompilationSubgraphFn()
91 const string& outside_compilation_attr_name, const string& xla_cluster_name,
|
| D | extract_outside_compilation_pass_test.cc | 241 const string &xla_cluster_name, const NameAttrList &func_name_attrs, in ExtractOutsideCompilationTest() argument
254 xla_cluster_attr_name, outside_compilation_attr_name, xla_cluster_name, 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
2998 clusters.emplace(xla_cluster_name, XlaClusterInfo{xla_cluster_name, func, n, in Run()
|