Searched refs:XlaClusterInfo (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.h | 86 struct XlaClusterInfo { struct
98 XlaClusterInfo() {} in XlaClusterInfo() function
99 XlaClusterInfo(const string& cluster_name, in XlaClusterInfo() function
|
| D | extract_outside_compilation_pass.h | 105 const std::unordered_map<string, XlaClusterInfo>& clusters, Graph* g,
|
| D | build_xla_ops_pass.cc | 217 struct XlaClusterInfo { struct
228 Status GetXlaClusterInfo(Node* n, XlaClusterInfo* result) { in GetXlaClusterInfo()
382 const XlaClusterInfo& cluster_info, in GetXlaRunArgs()
472 XlaClusterInfo cluster_info; in ReplaceNodeWithXlaCompileAndXlaRun()
|
| D | encapsulate_subgraphs_pass_test.cc | 515 std::unordered_map<string, XlaClusterInfo> clusters; in Encapsulate()
529 XlaClusterInfo{func, func_name_attrs, xla_computation_node, in Encapsulate()
|
| D | extract_outside_compilation_pass.cc | 2502 const std::unordered_map<string, XlaClusterInfo>& clusters, Graph* g, in ExtractOutsideCompilation()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.h | 67 std::unordered_map<string, XlaClusterInfo>* clusters, Graph* g,
|
| D | encapsulate_tpu_computations_pass.cc | 2884 std::unordered_map<string, XlaClusterInfo>* clusters, Graph* g, in ProcessHeadTailOutsideCompilation()
2978 std::unordered_map<string, XlaClusterInfo> clusters; in Run()
2998 clusters.emplace(xla_cluster_name, XlaClusterInfo{xla_cluster_name, func, n, in Run()
|