Home
last modified time | relevance | path

Searched refs:ClusterInfo (Results 1 – 3 of 3) 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
210 static string DumpInOutNames(const std::vector<ClusterInfo>& ci_vec) { in DumpInOutNames()
219 static void ClearCluster(ClusterInfo* cluster) { in ClearCluster()
481 ClusterInfo cluster; in TEST()
513 std::vector<ClusterInfo> ci_vec; in TEST()
550 ClusterInfo cluster; in TEST()
Dremote_fused_graph_execute_utils.cc127 string DumpCluster(const RemoteFusedGraphExecuteUtils::ClusterInfo& cluster) { in DumpCluster()
729 std::vector<ClusterInfo>* cluster_infos) { in ClusterizeNodes()
736 ClusterInfo ci; in ClusterizeNodes()
829 const ClusterInfo& cluster, const GraphDef& graph_def, in BuildClusterSubgraphDef()
888 ClusterInfo* cluster) { in BuildClusterByBorder()
945 const string& remote_fused_graph_node_name, const ClusterInfo& cluster, in FuseCluster()
1023 ClusterInfo graph_cluster; in FuseCluster()
1041 std::vector<ClusterInfo> ci_vec; in FuseRemoteGraphByNodeNames()
1064 ClusterInfo cluster; in FuseRemoteGraphByBorder()