Searched refs:node_to_stream_id (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_stream_util_test.cc | 40 std::unordered_map<int, int> node_to_stream_id; in TEST_F() local 43 status = gpu_stream_util::AssignStreams(nullptr, opts, &node_to_stream_id); in TEST_F() 48 status = gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id); in TEST_F() 52 status = gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id); in TEST_F() 60 std::unordered_map<int, int> node_to_stream_id; in TEST_F() local 62 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F() 63 EXPECT_EQ(2, node_to_stream_id.size()); // _SOURCE and _SINK in TEST_F() 72 std::unordered_map<int, int> node_to_stream_id; in TEST_F() local 74 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F() 77 EXPECT_EQ(5, node_to_stream_id.size()); in TEST_F() [all …]
|
D | gpu_stream_util.cc | 31 std::unordered_map<int, int>* node_to_stream_id) { in AssignStreams() argument 38 if (node_to_stream_id == nullptr) { in AssignStreams() 87 stream_id = (*node_to_stream_id)[e->src()->id()]; in AssignStreams() 102 (*node_to_stream_id)[node_id] = stream_id % opts.max_streams; in AssignStreams()
|
D | gpu_stream_util.h | 40 std::unordered_map<int, int>* node_to_stream_id);
|
D | gpu_device.cc | 466 std::unordered_map<int, int> node_to_stream_id; in FillContextMap() local 468 gpu_stream_util::AssignStreams(graph, opts, &node_to_stream_id)); in FillContextMap() 476 auto mapped_stream = node_to_stream_id[n->id()]; in FillContextMap() 479 VLOG(3) << "Assigned stream " << node_to_stream_id[n->id()] in FillContextMap()
|