| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 102 ScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in ScopedTracer()
134 StreamExecutor *stream_exec_;
143 MakeScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in MakeScopedTracer()
155 /* static */ mutex StreamExecutor::static_mu_{LINKER_INITIALIZED};
157 StreamExecutor::StreamExecutor(PlatformKind platform_kind, in StreamExecutor() function in stream_executor::StreamExecutor
180 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
205 StreamExecutor::~StreamExecutor() { in ~StreamExecutor()
224 port::Status StreamExecutor::Init(int device_ordinal, in Init()
230 port::Status StreamExecutor::Init() { in Init()
234 bool StreamExecutor::GetKernel(const MultiKernelLoaderSpec &spec, in GetKernel()
[all …]
|
| D | stream_test.cc | 25 std::unique_ptr<StreamExecutor> NewStreamExecutor() { in NewStreamExecutor()
34 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
40 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
47 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
68 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
99 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
138 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
|
| D | platform.h | 34 class StreamExecutor; variable
143 virtual port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) = 0;
148 virtual port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
153 virtual port::StatusOr<StreamExecutor*> GetExecutor(
159 virtual port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| D | executor_cache.h | 38 typedef port::StatusOr<std::unique_ptr<StreamExecutor>> ExecutorFactory();
39 port::StatusOr<StreamExecutor*> GetOrCreate(
45 port::StatusOr<StreamExecutor*> Get(const StreamExecutorConfig& config);
65 std::pair<StreamExecutorConfig, std::unique_ptr<StreamExecutor>>>
|
| D | stream_executor_pimpl.h | 71 class StreamExecutor {
73 explicit StreamExecutor(PlatformKind kind,
76 StreamExecutor(
80 ~StreamExecutor();
327 port::Status EnablePeerAccessTo(StreamExecutor *other);
335 bool CanEnablePeerAccessTo(StreamExecutor *other);
722 SE_DISALLOW_COPY_AND_ASSIGN(StreamExecutor);
728 explicit ScopedModuleHandle(StreamExecutor *executor, in ScopedModuleHandle()
754 StreamExecutor *executor_;
764 inline DeviceMemory<T> StreamExecutor::AllocateArray(uint64 element_count) { in AllocateArray()
[all …]
|
| D | timer.h | 29 class StreamExecutor; variable
41 explicit Timer(StreamExecutor *parent);
62 StreamExecutor *parent_;
|
| D | event.h | 30 class StreamExecutor; variable
49 explicit Event(StreamExecutor* stream_exec); // NOLINT
72 StreamExecutor* stream_exec_;
|
| D | device_memory.h | 44 class StreamExecutor; variable
97 friend class StreamExecutor;
202 ScopedDeviceMemory(StreamExecutor *parent, DeviceMemoryBase value);
205 ScopedDeviceMemory(StreamExecutor *parent,
267 StreamExecutor *parent_; // See constructor.
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.h | 41 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
48 Status TransferBufferToInfeed(se::StreamExecutor* executor, int64 size,
54 se::StreamExecutor* executor, int64 size, const void* source);
58 se::StreamExecutor* executor,
62 StatusOr<Shape> TransferArrayBufferFromOutfeed(se::StreamExecutor* executor,
70 se::StreamExecutor* executor,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 52 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in InspectNumaNodes()
76 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in DeviceToBus()
80 port::StatusOr<StreamExecutor*> ROCmPlatform::FirstExecutorForBus( in FirstExecutorForBus()
112 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
120 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
129 port::StatusOr<StreamExecutor*> ROCmPlatform::GetExecutor( in GetExecutor()
135 port::StatusOr<std::unique_ptr<StreamExecutor>>
137 auto executor = MakeUnique<StreamExecutor>( in GetUncachedExecutor()
|
| D | rocm_platform.h | 56 port::StatusOr<StreamExecutor*> FirstExecutorForBus(int bus_ordinal);
67 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
69 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
72 port::StatusOr<StreamExecutor*> GetExecutor(
75 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 106 const std::vector<se::StreamExecutor*>& stream_executors() const { in stream_executors()
111 StatusOr<se::StreamExecutor*> stream_executor(int device_ordinal) const;
116 se::StreamExecutor* default_stream_executor() const { in default_stream_executor()
125 StatusOr<StreamPool::Ptr> BorrowStream(se::StreamExecutor* executor);
160 absl::Span<se::StreamExecutor* const> stream_executors,
173 std::vector<se::StreamExecutor*> stream_executors_;
178 absl::flat_hash_map<se::StreamExecutor*, std::unique_ptr<StreamPool>>
|
| D | compiler.h | 149 std::unique_ptr<HloModule> module, se::StreamExecutor* executor,
156 absl::Span<se::StreamExecutor* const> executors,
170 std::unique_ptr<HloModule> module, se::StreamExecutor* executor,
178 std::vector<std::vector<se::StreamExecutor*>> stream_exec,
191 std::vector<std::vector<se::StreamExecutor*>> stream_exec,
202 se::StreamExecutor* executor) const;
212 se::StreamExecutor* executor) const;
|
| D | generic_transfer_manager.h | 53 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
55 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
59 Status ResetDevices(absl::Span<se::StreamExecutor* const> executors) override;
69 Status TransferLiteralFromDeviceInternal(se::StreamExecutor* executor,
|
| D | device_memory_allocator.cc | 29 absl::Span<se::StreamExecutor* const> stream_executors) in StreamExecutorMemoryAllocator()
35 TF_ASSIGN_OR_RETURN(se::StreamExecutor * stream_executor, in Allocate()
53 TF_ASSIGN_OR_RETURN(se::StreamExecutor * stream_executor, in Deallocate()
62 StatusOr<se::StreamExecutor*> StreamExecutorMemoryAllocator::GetStreamExecutor( in GetStreamExecutor()
|
| D | backend.cc | 135 StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) { in BorrowStream()
144 absl::Span<se::StreamExecutor* const> stream_executors, in Backend()
153 for (se::StreamExecutor* exec : stream_executors) { in Backend()
193 StatusOr<se::StreamExecutor*> Backend::stream_executor( in stream_executor()
216 TF_ASSIGN_OR_RETURN(se::StreamExecutor * executor_a, in devices_equivalent()
218 TF_ASSIGN_OR_RETURN(se::StreamExecutor * executor_b, in devices_equivalent()
|
| D | stream_pool_test.cc | 28 std::unique_ptr<se::StreamExecutor> NewStreamExecutor() { in NewStreamExecutor()
39 std::unique_ptr<se::StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
60 std::unique_ptr<se::StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
97 std::unique_ptr<se::StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
136 std::unique_ptr<se::StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 87 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in InspectNumaNodes()
111 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in DeviceToBus()
115 port::StatusOr<StreamExecutor*> CudaPlatform::FirstExecutorForBus( in FirstExecutorForBus()
146 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
154 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
163 port::StatusOr<StreamExecutor*> CudaPlatform::GetExecutor( in GetExecutor()
169 port::StatusOr<std::unique_ptr<StreamExecutor>>
171 auto executor = MakeUnique<StreamExecutor>( in GetUncachedExecutor()
|
| D | cuda_platform.h | 58 port::StatusOr<StreamExecutor*> FirstExecutorForBus(int bus_ordinal);
69 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
71 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
74 port::StatusOr<StreamExecutor*> GetExecutor(
77 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 47 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec,
51 absl::Span<se::StreamExecutor* const> executors,
55 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec,
59 std::vector<std::vector<se::StreamExecutor*>> stream_exec,
64 std::vector<std::vector<se::StreamExecutor*>> stream_exec,
|
| D | platform.cc | 46 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::ExecutorForDevice( in ExecutorForDevice()
55 port::StatusOr<StreamExecutor*>
65 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::GetExecutor( in GetExecutor()
71 port::StatusOr<std::unique_ptr<StreamExecutor>>
74 auto executor = absl::make_unique<StreamExecutor>( in GetUncachedExecutor()
|
| D | platform.h | 42 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
44 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
47 port::StatusOr<StreamExecutor*> GetExecutor(
50 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.h | 41 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
51 se::StreamExecutor* executor, int64 size, const void* source);
55 Status EnqueueBuffersToInfeed(se::StreamExecutor* executor,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 44 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
52 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
61 port::StatusOr<StreamExecutor*> HostPlatform::GetExecutor( in GetExecutor()
67 port::StatusOr<std::unique_ptr<StreamExecutor>>
69 auto executor = MakeUnique<StreamExecutor>( in GetUncachedExecutor()
|
| D | host_platform.h | 55 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
57 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
60 port::StatusOr<StreamExecutor*> GetExecutor(
63 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|