Searched refs:AssignStreams (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_stream_util_test.cc | 43 status = gpu_stream_util::AssignStreams(nullptr, opts, &node_to_stream_id); in TEST_F()
45 status = gpu_stream_util::AssignStreams(&g, opts, nullptr); 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()
62 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F()
74 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F()
94 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F()
125 TF_ASSERT_OK(gpu_stream_util::AssignStreams(&g, opts, &node_to_stream_id)); in TEST_F()
|
| D | gpu_stream_util.h | 39 Status AssignStreams(const Graph* graph, const AssignStreamsOpts& opts,
|
| D | gpu_stream_util.cc | 30 Status AssignStreams(const Graph* graph, const AssignStreamsOpts& opts, in AssignStreams() function
|
| D | gpu_device.cc | 468 gpu_stream_util::AssignStreams(graph, opts, &node_to_stream_id)); in FillContextMap()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_assignment_test.cc | 61 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()
82 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()
126 std::unique_ptr<StreamAssignment> assignment = AssignStreams(*module); in TEST_F()
|
| D | stream_assignment.h | 41 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module);
|
| D | gpu_hlo_schedule_test.cc | 85 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()
145 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()
215 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()
294 std::unique_ptr<StreamAssignment> streams = AssignStreams(*module); in TEST_F()
|
| D | stream_assignment.cc | 122 std::unique_ptr<StreamAssignment> AssignStreams(const HloModule& module) { in AssignStreams() function
|
| D | nvptx_compiler.cc | 659 std::unique_ptr<StreamAssignment> stream_assignment = AssignStreams(*module); in RunBackend()
|