Searched refs:stream_executors (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | device_memory_allocator.cc | 29 stream_executors) 31 stream_executors_(stream_executors.begin(), stream_executors.end()) {} 26 StreamExecutorMemoryAllocator( const perftools::gputools::Platform* platform, tensorflow::gtl::ArraySlice<perftools::gputools::StreamExecutor*> stream_executors) argument
|
H A D | platform_util.cc | 220 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); 226 thread_pool.Schedule([platform, i, &stream_executors]() { 234 stream_executors[i] = executor; 247 if (std::all_of(stream_executors.begin(), stream_executors.end(), 252 return stream_executors;
|
H A D | backend.cc | 82 TF_ASSIGN_OR_RETURN(auto stream_executors, 89 new Backend(platform, compiler, stream_executors, transfer_manager, 125 tensorflow::gtl::ArraySlice<se::StreamExecutor*> stream_executors, 133 for (se::StreamExecutor* exec : stream_executors) { 140 MakeUnique<StreamExecutorMemoryAllocator>(platform, stream_executors); 123 Backend( perftools::gputools::Platform* platform, Compiler* compiler, tensorflow::gtl::ArraySlice<se::StreamExecutor*> stream_executors, TransferManager* transfer_manager, ComputationPlacer* computation_placer, int intra_op_parallelism_threads) argument
|
H A D | device_memory_allocator.h | 69 stream_executors);
|
H A D | backend.h | 99 const std::vector<perftools::gputools::StreamExecutor*>& stream_executors() function in class:xla::Backend 162 stream_executors,
|
Completed in 767 milliseconds