Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_stream_util_test.cc40 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 …]
Dgpu_stream_util.cc31 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()
Dgpu_stream_util.h40 std::unordered_map<int, int>* node_to_stream_id);
Dgpu_device.cc466 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()