HomeSort by relevance Sort by last modified time
    Searched refs:StreamExecutorInterface (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
plugin_registry.h 35 class StreamExecutorInterface;
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*);
stream_executor_internal.h 147 class StreamExecutorInterface {
150 StreamExecutorInterface() {}
153 virtual ~StreamExecutorInterface() {}
157 virtual StreamExecutorInterface *GetUnderlyingExecutor() { return this; }
224 virtual port::Status EnablePeerAccessTo(StreamExecutorInterface *other) = 0;
225 virtual bool CanEnablePeerAccessTo(StreamExecutorInterface *other) = 0;
328 SE_DISALLOW_COPY_AND_ASSIGN(StreamExecutorInterface);
332 std::function<StreamExecutorInterface *(const PluginConfig &)>;
stream_executor_pimpl.h 78 std::unique_ptr<internal::StreamExecutorInterface> implementation);
395 internal::StreamExecutorInterface *implementation();
409 // implementation in StreamExecutorInterface::Launch().
595 std::unique_ptr<internal::StreamExecutorInterface> implementation_;
    [all...]
stream_executor_pimpl.cc 18 // implements StreamExecutorInterface.
62 internal::StreamExecutorInterface *StreamExecutorImplementationFromPlatformKind(
169 std::unique_ptr<internal::StreamExecutorInterface> implementation)
777 internal::StreamExecutorInterface *StreamExecutor::implementation() {
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.h 53 class InterpreterExecutor : public internal::StreamExecutorInterface {
170 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override {
174 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override {
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.h 16 // The CUDA implementation of the StreamExecutorInterface functionality.
44 class CUDAExecutor : public internal::StreamExecutorInterface {
159 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override;
161 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override;
cuda_rng.cc 284 [](gpu::internal::StreamExecutorInterface
cuda_gpu_executor.cc 732 bool CUDAExecutor::CanEnablePeerAccessTo(StreamExecutorInterface *other) {
737 port::Status CUDAExecutor::EnablePeerAccessTo(StreamExecutorInterface *other) {
    [all...]
cuda_fft.cc 554 [](gpu::internal::StreamExecutorInterface
cuda_blas.cc     [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.h 45 class HostExecutor : public internal::StreamExecutorInterface {
152 port::Status EnablePeerAccessTo(StreamExecutorInterface *other) override {
156 bool CanEnablePeerAccessTo(StreamExecutorInterface *other) override {

Completed in 789 milliseconds