Searched refs:StreamExecutorInterface (Results 1 – 21 of 21) sorted by relevance
34 class StreamExecutorInterface; variable52 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);53 typedef dnn::DnnSupport* (*DnnFactory)(internal::StreamExecutorInterface*);54 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*);55 typedef rng::RngSupport* (*RngFactory)(internal::StreamExecutorInterface*);
165 class StreamExecutorInterface {168 StreamExecutorInterface() {} in StreamExecutorInterface() function171 virtual ~StreamExecutorInterface() {} in ~StreamExecutorInterface()175 virtual StreamExecutorInterface *GetUnderlyingExecutor() { return this; } in GetUnderlyingExecutor()267 virtual port::Status EnablePeerAccessTo(StreamExecutorInterface *other) = 0;268 virtual bool CanEnablePeerAccessTo(StreamExecutorInterface *other) = 0;386 SE_DISALLOW_COPY_AND_ASSIGN(StreamExecutorInterface);
23 bool StreamExecutorInterface::HostCallback(Stream* stream, in HostCallback()
77 std::unique_ptr<internal::StreamExecutorInterface> implementation,446 internal::StreamExecutorInterface *implementation();675 std::unique_ptr<internal::StreamExecutorInterface> implementation_;
141 std::unique_ptr<internal::StreamExecutorInterface> implementation, in StreamExecutor()881 internal::StreamExecutorInterface *StreamExecutor::implementation() { in implementation()
53 using StreamExecutorInterface = variable54 ::stream_executor::internal::StreamExecutorInterface;193 Status EnablePeerAccessTo(StreamExecutorInterface* other) override { in EnablePeerAccessTo()196 bool CanEnablePeerAccessTo(StreamExecutorInterface* other) override { in CanEnablePeerAccessTo()
35 : public stream_executor::internal::StreamExecutorInterface {
44 class HostExecutor : public internal::StreamExecutorInterface {143 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()147 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
51 class XlaInterpreterExecutor : public internal::StreamExecutorInterface {176 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()180 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
44 class GpuExecutor : public internal::StreamExecutorInterface {187 port::Status EnablePeerAccessTo(StreamExecutorInterface* other) override;189 bool CanEnablePeerAccessTo(StreamExecutorInterface* other) override;
235 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_curand()
805 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()810 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
569 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_cufft()
4655 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_cudnn()
3761 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
237 class CStreamExecutor : public internal::StreamExecutorInterface {587 port::Status EnablePeerAccessTo(StreamExecutorInterface* other) override { in EnablePeerAccessTo()591 bool CanEnablePeerAccessTo(StreamExecutorInterface* other) override { in CanEnablePeerAccessTo()
292 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_rocrand()
713 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()718 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
598 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_rocfft()
2693 [](internal::StreamExecutorInterface *parent) in initialize_rocblas()
5086 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_miopen()