Searched refs:StreamExecutorInterface (Results 1 – 18 of 18) sorted by relevance
35 class StreamExecutorInterface; variable53 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*);
166 class StreamExecutorInterface {169 StreamExecutorInterface() {} in StreamExecutorInterface() function172 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 &)>;
48 bool StreamExecutorInterface::HostCallback(Stream* stream, in HostCallback()
78 std::unique_ptr<internal::StreamExecutorInterface> implementation);437 internal::StreamExecutorInterface *implementation();649 std::unique_ptr<internal::StreamExecutorInterface> implementation_;
63 internal::StreamExecutorInterface *StreamExecutorImplementationFromPlatformKind( in StreamExecutorImplementationFromPlatformKind()182 std::unique_ptr<internal::StreamExecutorInterface> implementation) in StreamExecutor()893 internal::StreamExecutorInterface *StreamExecutor::implementation() { in implementation()
52 class XlaInterpreterExecutor : public internal::StreamExecutorInterface {170 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()174 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
44 class HostExecutor : public internal::StreamExecutorInterface {152 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override { in EnablePeerAccessTo()156 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override { in CanEnablePeerAccessTo()
44 class GpuExecutor : public internal::StreamExecutorInterface {186 port::Status EnablePeerAccessTo(StreamExecutorInterface* other) override;188 bool CanEnablePeerAccessTo(StreamExecutorInterface* other) override;
235 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_curand()
833 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()838 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
540 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_cufft()
4122 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_cudnn()
2803 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
293 [](internal::StreamExecutorInterface* parent) -> rng::RngSupport* { in initialize_rocrand()
714 bool GpuExecutor::CanEnablePeerAccessTo(StreamExecutorInterface* other) { in CanEnablePeerAccessTo()719 port::Status GpuExecutor::EnablePeerAccessTo(StreamExecutorInterface* other) { in EnablePeerAccessTo()
593 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_rocfft()
2340 [](internal::StreamExecutorInterface *parent) in initialize_rocblas()
4490 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_miopen()