Home
last modified time | relevance | path

Searched refs:ci_vec (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc210 static string DumpInOutNames(const std::vector<ClusterInfo>& ci_vec) { in DumpInOutNames() argument
211 for (int i = 0; i < ci_vec.size(); ++i) { in DumpInOutNames()
213 LOG(INFO) << "input: " << IterToString(std::get<1>(ci_vec.at(i))); in DumpInOutNames()
214 LOG(INFO) << "output: " << IterToString(std::get<2>(ci_vec.at(i))); in DumpInOutNames()
513 std::vector<ClusterInfo> ci_vec; in TEST() local
515 RemoteFusedGraphExecuteUtils::ClusterizeNodes({"J"}, graph_def, &ci_vec)); in TEST()
516 ASSERT_EQ(1, ci_vec.size()); in TEST()
517 EXPECT_EQ(2, std::get<1>(ci_vec.at(0)).size()) << DumpInOutNames(ci_vec); in TEST()
518 EXPECT_EQ(1, std::get<2>(ci_vec.at(0)).size()) << DumpInOutNames(ci_vec); in TEST()
520 ci_vec.clear(); in TEST()
[all …]
Dremote_fused_graph_execute_utils.cc1041 std::vector<ClusterInfo> ci_vec; in FuseRemoteGraphByNodeNames() local
1043 subgraph_nodes, input_graph_def, &ci_vec)); in FuseRemoteGraphByNodeNames()
1045 for (size_t i = 0; i < ci_vec.size(); ++i) { in FuseRemoteGraphByNodeNames()
1049 remote_fused_graph_node_name, ci_vec.at(i), in FuseRemoteGraphByNodeNames()