HomeSort by relevance Sort by last modified time
    Searched refs:node_to_stream_id (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
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...]
gpu_stream_util.h 40 std::unordered_map<int, int>* node_to_stream_id);
gpu_stream_util.cc 31 std::unordered_map<int, int>* node_to_stream_id) {
38 if (node_to_stream_id == nullptr) {
40 errors::InvalidArgument("Bad node_to_stream_id argument supplied."));
87 stream_id = (*node_to_stream_id)[e->src()->id()];
102 (*node_to_stream_id)[node_id] = stream_id % opts.max_streams;
gpu_device.cc 384 std::unordered_map<int, int> node_to_stream_id; local
386 gpu_stream_util::AssignStreams(graph, opts, &node_to_stream_id));
394 auto mapped_stream = node_to_stream_id[n->id()];
397 VLOG(3) << "Assigned stream " << node_to_stream_id[n->id()]
    [all...]

Completed in 558 milliseconds