Home
last modified time | relevance | path

Searched refs:StreamExecutorInterface (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h35 class StreamExecutorInterface; variable
53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
54 typedef dnn::DnnSupport* (*DnnFactory)(internal::StreamExecutorInterface*);
55 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*);
56 typedef rng::RngSupport* (*RngFactory)(internal::StreamExecutorInterface*);
Dstream_executor_internal.h166 class StreamExecutorInterface {
169 StreamExecutorInterface() {} in StreamExecutorInterface() function
172 virtual ~StreamExecutorInterface() {} in ~StreamExecutorInterface()
176 virtual StreamExecutorInterface *GetUnderlyingExecutor() { return this; } in GetUnderlyingExecutor()
263 virtual port::Status EnablePeerAccessTo(StreamExecutorInterface *other) = 0;
264 virtual bool CanEnablePeerAccessTo(StreamExecutorInterface *other) = 0;
378 SE_DISALLOW_COPY_AND_ASSIGN(StreamExecutorInterface);
382 std::function<StreamExecutorInterface *(const PluginConfig &)>;
Dstream_executor_internal.cc48 bool StreamExecutorInterface::HostCallback(Stream* stream, in HostCallback()
Dstream_executor_pimpl.h78 std::unique_ptr<internal::StreamExecutorInterface> implementation);
437 internal::StreamExecutorInterface *implementation();
649 std::unique_ptr<internal::StreamExecutorInterface> implementation_;
Dstream_executor_pimpl.cc63 internal::StreamExecutorInterface *StreamExecutorImplementationFromPlatformKind( in StreamExecutorImplementationFromPlatformKind()
182 std::unique_ptr<internal::StreamExecutorInterface> implementation) in StreamExecutor()
893 internal::StreamExecutorInterface *StreamExecutor::implementation() { in implementation()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h52 class XlaInterpreterExecutor : public internal::StreamExecutorInterface {
170 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()
174 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h44 class HostExecutor : public internal::StreamExecutorInterface {
152 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()
156 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h44 class GpuExecutor : public internal::StreamExecutorInterface {
186 port::Status EnablePeerAccessTo(StreamExecutorInterface* other) override;
188 bool CanEnablePeerAccessTo(StreamExecutorInterface* other) override;
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_rng.cc235 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_curand()
Dcuda_gpu_executor.cc833 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()
838 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
Dcuda_fft.cc540 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_cufft()
Dcuda_dnn.cc4122 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_cudnn()
Dcuda_blas.cc2803 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc293 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_rocrand()
Drocm_gpu_executor.cc714 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()
719 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
Drocm_fft.cc593 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_rocfft()
Drocm_blas.cc2340 [](internal::StreamExecutorInterface *parent) in initialize_rocblas()
Drocm_dnn.cc4490 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_miopen()