Searched refs:stream_executors (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 109 auto stream_executors, in CreateBackend()
116 new Backend(platform, compiler, stream_executors, transfer_manager, in CreateBackend()
144 absl::Span<se::StreamExecutor* const> stream_executors, in Backend() argument
153 for (se::StreamExecutor* exec : stream_executors) { in Backend()
160 platform, stream_executors); in Backend()
|
| D | device_memory_allocator.cc | 29 absl::Span<se::StreamExecutor* const> stream_executors) in StreamExecutorMemoryAllocator() argument
31 stream_executors_(stream_executors.begin(), stream_executors.end()) {} in StreamExecutorMemoryAllocator()
|
| D | platform_util.cc | 228 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors() local
245 thread_pool.Schedule([platform, i, &stream_executors]() { in GetStreamExecutors()
253 stream_executors[i] = executor; in GetStreamExecutors()
266 if (absl::c_all_of(stream_executors, in GetStreamExecutors()
271 return stream_executors; in GetStreamExecutors()
|
| D | backend.h | 106 const std::vector<se::StreamExecutor*>& stream_executors() const { in stream_executors() function
160 absl::Span<se::StreamExecutor* const> stream_executors,
|
| D | device_memory_allocator.h | 83 absl::Span<se::StreamExecutor* const> stream_executors);
|
| D | service.cc | 171 auto stream_executors = execute_backend_->stream_executors(); in Service() local
173 se::StreamExecutor* executor = stream_executors.at(i); in Service()
232 absl::Span<se::StreamExecutor* const> stream_executors) const { in ResolveAndValidateArguments()
233 CHECK_EQ(options_.number_of_replicas(), stream_executors.size()); in ResolveAndValidateArguments()
247 int replica_device_ordinal = stream_executors[replica]->device_ordinal(); in ResolveAndValidateArguments()
|
| D | service.h | 219 absl::Span<se::StreamExecutor* const> stream_executors) const;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 50 for (auto* executor : client_->backend().stream_executors()) { in ~XlaCompilationCache()
|
| D | xla_device.cc | 106 backend->stream_executors()[device_ordinal]); in GetOrCreateXlaDeviceAllocator()
|