Searched refs:stream_executors (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 87 auto stream_executors, in CreateBackend()
94 new Backend(platform, compiler, stream_executors, transfer_manager, in CreateBackend()
122 absl::Span<se::StreamExecutor* const> stream_executors, in Backend() argument
130 stream_executors_(stream_executors.begin(), stream_executors.end()) { in Backend()
|
| D | platform_util.cc | 182 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors() local
199 thread_pool.Schedule([platform, i, &stream_executors]() { in GetStreamExecutors()
207 stream_executors[i] = executor; in GetStreamExecutors()
222 for (se::StreamExecutor* executor : 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 | service.cc | 173 auto stream_executors = execute_backend_->stream_executors(); in Service() local
175 se::StreamExecutor* executor = stream_executors.at(i); in Service()
234 absl::Span<se::StreamExecutor* const> stream_executors) const { in ResolveAndValidateArguments()
235 CHECK_EQ(options_.number_of_replicas(), stream_executors.size()); in ResolveAndValidateArguments()
|
| D | service.h | 228 absl::Span<se::StreamExecutor* const> stream_executors) const;
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_memory_allocator.h | 242 absl::Span<StreamExecutor *const> stream_executors);
|
| D | stream_executor_pimpl.cc | 893 absl::Span<StreamExecutor *const> stream_executors) in StreamExecutorMemoryAllocator() argument
895 stream_executors_(stream_executors.begin(), stream_executors.end()) {} in StreamExecutorMemoryAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 83 platform_, backend_->stream_executors()); in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 126 backend->stream_executors()[device_ordinal]); in GetOrCreateXlaDeviceAllocator()
|
| D | xla_compilation_cache.cc | 64 for (auto* executor : client_->backend().stream_executors()) { in ~XlaCompilationCache()
|