Home
last modified time | relevance | path

Searched refs:KernelInterface (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h89 class KernelInterface {
92 KernelInterface() {} in KernelInterface() function
95 virtual ~KernelInterface() {} in ~KernelInterface()
107 SE_DISALLOW_COPY_AND_ASSIGN(KernelInterface);
359 virtual std::unique_ptr<KernelInterface> CreateKernelImplementation() = 0;
386 using KernelFactory = std::function<KernelInterface*()>;
Dkernel.h92 class KernelInterface; variable
146 KernelBase(StreamExecutor *parent, internal::KernelInterface *implementation);
161 const internal::KernelInterface *implementation() const { in implementation()
167 internal::KernelInterface *implementation() { return implementation_.get(); } in implementation()
189 std::unique_ptr<internal::KernelInterface> implementation_;
506 TypedKernel(StreamExecutor *parent, internal::KernelInterface *implementation)
Dkernel.cc75 internal::KernelInterface *implementation) in KernelBase()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_kernel.h36 class GpuKernel : public internal::KernelInterface {
Dgpu_executor.h227 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc846 std::unique_ptr<internal::KernelInterface>
848 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc944 std::unique_ptr<internal::KernelInterface>
946 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()