Searched refs:oc_cluster_name (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 108 Graph* g, const string& oc_cluster_name, in BuildRecvAtHostNode() argument
111 absl::StrCat("outside_compilation_", oc_cluster_name, "_recv"), in BuildRecvAtHostNode()
121 "key", absl::StrCat("host_compute_channel_", oc_cluster_name)); in BuildRecvAtHostNode()
133 Graph* g, const string& oc_cluster_name, in ReplaceArgNodesWithRecvAtHostNode() argument
141 BuildRecvAtHostNode(g, oc_cluster_name, *recv_at_host_dtypes, in ReplaceArgNodesWithRecvAtHostNode()
209 Graph* g, const string& oc_cluster_name, in BuildSendFromHostNode() argument
213 absl::StrCat("outside_compilation_", oc_cluster_name, "_send"), in BuildSendFromHostNode()
223 "key", absl::StrCat("host_compute_channel_", oc_cluster_name)); in BuildSendFromHostNode()
250 Graph* g, const string& oc_cluster_name, in ReplaceRetNodesWithSendFromHostNode() argument
258 BuildSendFromHostNode(g, oc_cluster_name, ret_nodes, in ReplaceRetNodesWithSendFromHostNode()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3516 const string& oc_cluster_name = oc_cluster_iter.first; in ReplicateOutsideCompilationNodes() local
3520 int core = host_compute_core.at(oc_cluster_name); in ReplicateOutsideCompilationNodes()
3908 const string& oc_cluster_name = iter.first; in ParseHostComputeCores() local
3909 if (host_compute_core->find(oc_cluster_name) == host_compute_core->end()) { in ParseHostComputeCores()
3911 (*host_compute_core)[oc_cluster_name] = 0; in ParseHostComputeCores()
|