Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h88 class KernelInterface {
91 KernelInterface() {} in KernelInterface() function
94 virtual ~KernelInterface() {} in ~KernelInterface()
106 SE_DISALLOW_COPY_AND_ASSIGN(KernelInterface);
362 virtual std::unique_ptr<KernelInterface> CreateKernelImplementation() = 0;
Dkernel.h93 class KernelInterface; variable
147 KernelBase(StreamExecutor *parent, internal::KernelInterface *implementation);
162 const internal::KernelInterface *implementation() const { in implementation()
168 internal::KernelInterface *implementation() { return implementation_.get(); } in implementation()
190 std::unique_ptr<internal::KernelInterface> implementation_;
517 TypedKernel(StreamExecutor *parent, internal::KernelInterface *implementation)
Dkernel.cc74 internal::KernelInterface *implementation) in KernelBase()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_kernel.h36 class GpuKernel : public internal::KernelInterface {
Dgpu_executor.h225 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h166 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h189 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h176 std::unique_ptr<stream_executor::internal::KernelInterface>
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc780 std::unique_ptr<internal::KernelInterface>
782 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc875 std::unique_ptr<internal::KernelInterface>
877 return std::unique_ptr<internal::KernelInterface>(new GpuKernel()); in CreateKernelImplementation()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc646 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation()