Home
last modified time | relevance | path

Searched refs:ClusterInfo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h94 using ClusterInfo = std::tuple<std::unordered_set<string>, // node names variable
199 std::vector<ClusterInfo>* cluster_infos);
202 static Status BuildClusterSubgraphDef(const ClusterInfo& cluster,
211 ClusterInfo* cluster);
223 const ClusterInfo& cluster,
Dremote_fused_graph_execute_utils_test.cc29 using ClusterInfo = RemoteFusedGraphExecuteUtils::ClusterInfo; typedef
209 static string DumpInOutNames(const std::vector<ClusterInfo>& ci_vec) { in DumpInOutNames()
218 static void ClearCluster(ClusterInfo* cluster) { in ClearCluster()
480 ClusterInfo cluster; in TEST()
512 std::vector<ClusterInfo> ci_vec; in TEST()
549 ClusterInfo cluster; in TEST()
Dremote_fused_graph_execute_utils.cc127 string DumpCluster(const RemoteFusedGraphExecuteUtils::ClusterInfo& cluster) { in DumpCluster()
730 std::vector<ClusterInfo>* cluster_infos) { in ClusterizeNodes()
737 ClusterInfo ci; in ClusterizeNodes()
831 const ClusterInfo& cluster, const GraphDef& graph_def, in BuildClusterSubgraphDef()
890 ClusterInfo* cluster) { in BuildClusterByBorder()
947 const string& remote_fused_graph_node_name, const ClusterInfo& cluster, in FuseCluster()
1025 ClusterInfo graph_cluster; in FuseCluster()
1043 std::vector<ClusterInfo> ci_vec; in FuseRemoteGraphByNodeNames()
1066 ClusterInfo cluster; in FuseRemoteGraphByBorder()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc323 struct ClusterInfo { struct
346 absl::string_view name, const ClusterInfo& info) { in ClusterInfoToProtobuf()
355 absl::flat_hash_map<absl::string_view, ClusterInfo> cluster_name_to_info; in GetXlaAutoClusteringSummary()
364 ClusterInfo* info = &cluster_name_to_info[*cluster_name]; in GetXlaAutoClusteringSummary()