gpu_stream_util_test.cc | 40 std::unordered_map<int, int> node_to_stream_id; local 43 status = gpu_stream_util::AssignStreams(nullptr, opts, &node_to_stream_id); 48 status = gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id); 52 status = gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id); 60 std::unordered_map<int, int> node_to_stream_id; local 62 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); 63 EXPECT_EQ(2, node_to_stream_id.size()); // _SOURCE and _SINK 72 std::unordered_map<int, int> node_to_stream_id; local 74 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); 77 EXPECT_EQ(5, node_to_stream_id.size()) 91 std::unordered_map<int, int> node_to_stream_id; local 118 std::unordered_map<int, int> node_to_stream_id; local [all...] |