Searched refs:oc_cluster (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 400 const string& new_func_name, const string& oc_cluster, in RecvAtHost() argument
405 new_func_name, "_", oc_cluster); in RecvAtHost()
407 new_func_name, "_", oc_cluster, "_recv"); in RecvAtHost()
415 .WithAttr("_outside", oc_cluster) in RecvAtHost()
420 const string& new_func_name, const string& oc_cluster, in SendFromHost() argument
425 new_func_name, "_", oc_cluster); in SendFromHost()
427 new_func_name, "_", oc_cluster, "_send"); in SendFromHost()
440 .WithAttr("_outside", oc_cluster) in SendFromHost()
|
| D | extract_outside_compilation_pass.cc | 457 string oc_cluster; in LiftedArgsAndOutsideCompilationNodesInFunctionBody() local
460 &oc_cluster) in LiftedArgsAndOutsideCompilationNodesInFunctionBody()
462 TF_RET_CHECK(outside_compilation_attr_to_node.find(oc_cluster) != in LiftedArgsAndOutsideCompilationNodesInFunctionBody()
465 n, outside_compilation_attr_to_node.at(oc_cluster)); in LiftedArgsAndOutsideCompilationNodesInFunctionBody()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 376 const string& oc_cluster = outside_compilation_attr->s(); in FindTaggedNodes() local
377 if (oc_cluster.empty()) { in FindTaggedNodes()
399 iter->second[oc_cluster].push_back(node); in FindTaggedNodes()
|