| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 75 ScopedTracer(StreamExecutor* stream_exec, BeginCallT begin_call, in ScopedTracer()
107 StreamExecutor* stream_exec_;
116 MakeScopedTracer(StreamExecutor* stream_exec, BeginCallT begin_call, in MakeScopedTracer()
128 /* static */ absl::Mutex StreamExecutor::static_mu_{absl::kConstInit};
139 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
167 StreamExecutor::~StreamExecutor() { in ~StreamExecutor()
186 port::Status StreamExecutor::Init(DeviceOptions device_options) { in Init()
190 port::Status StreamExecutor::Init() { return Init(DeviceOptions::Default()); } in Init()
192 port::Status StreamExecutor::GetKernel(const MultiKernelLoaderSpec& spec, in GetKernel()
197 void StreamExecutor::UnloadKernel(const KernelBase* kernel) { in UnloadKernel()
[all …]
|
| D | stream_test.cc | 24 std::unique_ptr<StreamExecutor> NewStreamExecutor() { in NewStreamExecutor()
32 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
38 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
45 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
66 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
97 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
136 std::unique_ptr<StreamExecutor> executor = NewStreamExecutor(); in TEST_F()
|
| 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 | platform.h | 34 class StreamExecutor; variable
157 virtual port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) = 0;
162 virtual port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
167 virtual port::StatusOr<StreamExecutor*> GetExecutor(
173 virtual port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 34 TEST(StreamExecutor, SuccessfulRegistration) { in TEST() argument
51 port::StatusOr<StreamExecutor*> maybe_executor = in TEST()
56 TEST(StreamExecutor, NameNotSet) { in TEST() argument
71 TEST(StreamExecutor, InvalidNameWithSemicolon) { in TEST() argument
88 TEST(StreamExecutor, InvalidNameWithSlash) { in TEST() argument
104 TEST(StreamExecutor, CreateDeviceNotSet) { in TEST() argument
120 TEST(StreamExecutor, UnifiedMemoryAllocateNotSet) { in TEST() argument
149 StreamExecutor* GetExecutor(int ordinal) { in GetExecutor()
155 port::StatusOr<StreamExecutor*> maybe_executor = in GetExecutor()
182 StreamExecutor* executor = GetExecutor(0); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 68 Compiler* compiler, se::StreamExecutor* executor) const override;
87 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
93 virtual GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) = 0;
96 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
106 se::StreamExecutor* stream_exec, const CompileOptions& options,
122 HloModule* hlo_module, se::StreamExecutor* stream_exec,
127 se::StreamExecutor* stream_exec,
131 HloModule* hlo_module, se::StreamExecutor* stream_exec,
145 se::StreamExecutor* stream_exec, bool relocatable,
151 se::StreamExecutor* stream_exec, in LinkModules()
[all …]
|
| D | nvptx_compiler.h | 41 HloModule* hlo_module, se::StreamExecutor* stream_exec,
45 HloModule* hlo_module, se::StreamExecutor* stream_exec,
50 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
54 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable,
59 se::StreamExecutor* stream_exec,
77 se::StreamExecutor* stream_exec, const std::string& ptx,
|
| D | amdgpu_compiler.h | 37 HloModule* hlo_module, se::StreamExecutor* stream_exec,
41 HloModule* hlo_module, se::StreamExecutor* stream_exec,
44 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
48 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable,
|
| D | gpu_transfer_manager.h | 39 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
41 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
82 void EnsurePinnedBuffersAllocated(se::StreamExecutor* executor)
92 se::StreamExecutor* pinned_chunk_se_ ABSL_GUARDED_BY(mu_) = nullptr;
|
| D | xla_executor_state.h | 31 explicit GpuExecutorXLAState(stream_executor::StreamExecutor *) {} in GpuExecutorXLAState() argument
33 InfeedManager *getOrCreateInfeedManager(stream_executor::StreamExecutor *se) { in getOrCreateInfeedManager()
40 stream_executor::StreamExecutor *se) { in getOrCreateOutfeedManager()
49 T *getOrCreate(stream_executor::StreamExecutor *se, std::unique_ptr<T> &ptr) { in getOrCreate()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_defn.h | 31 class StreamExecutor; variable
39 stream_executor::StreamExecutor* executor;
43 explicit SE_Stream(stream_executor::StreamExecutor* parent) in SE_Stream()
49 explicit SE_Event(stream_executor::StreamExecutor* parent) : event(parent) {} in SE_Event()
54 explicit SE_Timer(stream_executor::StreamExecutor* parent) : timer(parent) {} in SE_Timer()
|
| D | tpu_transfer_manager.h | 54 Status TransferLiteralToInfeed(stream_executor::StreamExecutor* executor,
58 stream_executor::StreamExecutor* executor,
62 se::StreamExecutor* executor,
66 absl::Span<stream_executor::StreamExecutor* const> executor) override;
74 stream_executor::StreamExecutor* executor,
78 se::StreamExecutor* executor,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 48 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in InspectNumaNodes()
72 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in DeviceToBus()
76 port::StatusOr<StreamExecutor*> ROCmPlatform::FirstExecutorForBus( in FirstExecutorForBus()
113 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
121 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
130 port::StatusOr<StreamExecutor*> ROCmPlatform::GetExecutor( in GetExecutor()
142 port::StatusOr<std::unique_ptr<StreamExecutor>>
144 auto executor = std::make_unique<StreamExecutor>( in GetUncachedExecutor()
|
| D | rocm_platform.h | 55 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/ |
| D | compiler.h | 69 Compiler* compiler, se::StreamExecutor* executor) const { in LoadExecutable()
143 se::StreamExecutor* executor() const { return executor_; } in executor()
144 void set_executor(se::StreamExecutor* executor) { executor_ = executor; } in set_executor()
187 se::StreamExecutor* executor_ = nullptr;
244 std::unique_ptr<HloModule> module, se::StreamExecutor* executor,
247 std::unique_ptr<HloModule> module, se::StreamExecutor* executor, in RunHloPasses()
271 std::unique_ptr<HloModule> module, se::StreamExecutor* executor,
274 std::unique_ptr<HloModule> module, se::StreamExecutor* executor, in RunBackend()
295 std::vector<std::vector<se::StreamExecutor*>> stream_exec,
299 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in Compile()
[all …]
|
| D | backend.h | 107 const std::vector<se::StreamExecutor*>& stream_executors() const { in stream_executors()
112 StatusOr<se::StreamExecutor*> stream_executor(int device_ordinal) const;
117 se::StreamExecutor* default_stream_executor() const { in default_stream_executor()
126 StatusOr<StreamPool::Ptr> BorrowStream(se::StreamExecutor* executor);
161 absl::Span<se::StreamExecutor* const> stream_executors,
174 std::vector<se::StreamExecutor*> stream_executors_;
179 absl::flat_hash_map<se::StreamExecutor*, std::unique_ptr<StreamPool>>
|
| D | platform_util.cc | 127 static bool IsDeviceSupported(se::StreamExecutor* executor) { in IsDeviceSupported()
156 /* static */ StatusOr<std::vector<se::StreamExecutor*>>
174 std::vector<se::StreamExecutor*> stream_executors(device_count, nullptr); in GetStreamExecutors()
180 std::vector<se::StreamExecutor*>& stream_executors, in GetStreamExecutors()
185 se::StreamExecutor* executor = executor_status.ValueOrDie(); in GetStreamExecutors()
225 std::vector<se::StreamExecutor*> out; in GetStreamExecutors()
226 for (se::StreamExecutor* executor : stream_executors) { in GetStreamExecutors()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_platform.cc | 83 StreamExecutor* exec = *ExecutorForDevice(i); in InspectNumaNodes()
105 StreamExecutor* exec = *ExecutorForDevice(device_ordinal); in DeviceToBus()
109 port::StatusOr<StreamExecutor*> CudaPlatform::FirstExecutorForBus( in FirstExecutorForBus()
143 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
151 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
160 port::StatusOr<StreamExecutor*> CudaPlatform::GetExecutor( in GetExecutor()
172 port::StatusOr<std::unique_ptr<StreamExecutor>>
174 auto executor = std::make_unique<StreamExecutor>( in GetUncachedExecutor()
|
| D | cuda_platform.h | 56 port::StatusOr<StreamExecutor*> FirstExecutorForBus(int bus_ordinal);
70 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
72 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
75 port::StatusOr<StreamExecutor*> GetExecutor(
78 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.h | 40 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
42 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
46 se::StreamExecutor* executor, in CanShapedBufferBeAccessedNow()
52 se::StreamExecutor* executor, in CanBufferBeAccessedNow()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_platform.cc | 48 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice()
56 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDeviceWithPluginConfig( in ExecutorForDeviceWithPluginConfig()
65 port::StatusOr<StreamExecutor*> HostPlatform::GetExecutor( in GetExecutor()
71 port::StatusOr<std::unique_ptr<StreamExecutor>>
73 auto executor = std::make_unique<StreamExecutor>( in GetUncachedExecutor()
|
| D | host_platform.h | 56 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
58 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
61 port::StatusOr<StreamExecutor*> GetExecutor(
64 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.h | 35 class StreamExecutor; variable
87 se::StreamExecutor* const exec_;
99 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
153 EventMgr* GetEventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
160 std::map<se::StreamExecutor*, EventMgr*> event_mgr_map_ TF_GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 49 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::ExecutorForDevice( in ExecutorForDevice()
58 port::StatusOr<StreamExecutor*>
68 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::GetExecutor( in GetExecutor()
74 port::StatusOr<std::unique_ptr<StreamExecutor>>
77 auto executor = std::make_unique<StreamExecutor>( in GetUncachedExecutor()
|
| D | platform.h | 46 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
48 port::StatusOr<StreamExecutor*> ExecutorForDeviceWithPluginConfig(
51 port::StatusOr<StreamExecutor*> GetExecutor(
54 port::StatusOr<std::unique_ptr<StreamExecutor>> GetUncachedExecutor(
|