Searched refs:kXlaClusterIdAttr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 46 if (!TryGetNodeAttr(n->attrs(), kXlaClusterIdAttr, &name)) continue; in IsCpuGpuCompile()
178 AddNodeAttr(kXlaClusterIdAttr, call_def->name(), call_def); in RewriteSubgraph()
213 e->src()->attrs().Find(kXlaClusterIdAttr) != nullptr && in Encapsulate()
214 e->dst()->attrs().Find(kXlaClusterIdAttr) == nullptr && in Encapsulate()
230 kXlaClusterIdAttr, **graph, RewriteSubgraph, in Encapsulate()
243 const string& name = GetNodeAttrString(n->attrs(), kXlaClusterIdAttr); in BuildXlaLaunchOps()
|
| D | defs.cc | 31 const char* const kXlaClusterIdAttr = "_xla_compile_id"; variable
|
| D | defs.h | 39 extern const char* const kXlaClusterIdAttr; // "_xla_compile_id"
|
| D | encapsulate_xla_computations_pass_test.cc | 59 .Attr(kXlaClusterIdAttr, "launch0") in MakeOuterGraph()
110 node->AddAttr(kXlaClusterIdAttr, "launch0"); in MakeBodyGraph()
158 node->AddAttr(kXlaClusterIdAttr, "launch0"); in TEST()
218 node->AddAttr(kXlaClusterIdAttr, "launch0"); in TEST()
|
| D | compilability_check_util.cc | 722 return node.attr().find(kXlaClusterIdAttr) != node.attr().end() || in NodeCanTriggerXlaCompilation()
|