Searched refs:stream_executors (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddevice_memory_allocator.cc29 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 Dplatform_util.cc220 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 Dbackend.cc82 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 Ddevice_memory_allocator.h69 stream_executors);
H A Dbackend.h99 const std::vector<perftools::gputools::StreamExecutor*>& stream_executors() function in class:xla::Backend
162 stream_executors,

Completed in 767 milliseconds