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

  /external/tensorflow/tensorflow/stream_executor/
stream_executor_internal.h 71 class KernelInterface {
74 KernelInterface() {}
77 virtual ~KernelInterface() {}
89 SE_DISALLOW_COPY_AND_ASSIGN(KernelInterface);
315 virtual std::unique_ptr<KernelInterface> CreateKernelImplementation() = 0;
336 using KernelFactory = std::function<KernelInterface*()>;
kernel.cc 74 internal::KernelInterface *implementation)
kernel.h 94 class KernelInterface;
147 // Test-only constructor that can take a mock KernelInterface implementation.
148 KernelBase(StreamExecutor *parent, internal::KernelInterface *implementation);
163 const internal::KernelInterface *implementation() const {
169 internal::KernelInterface *implementation() { return implementation_.get(); }
191 std::unique_ptr<internal::KernelInterface> implementation_;
506 // Test-only constructor that can take a mock KernelInterface implementation.
508 TypedKernel(StreamExecutor *parent, internal::KernelInterface *implementation)
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_kernel.h 47 // Wraps a CUfunction to implement the platform-independent KernelInterface.
48 class CUDAKernel : public internal::KernelInterface {
cuda_gpu_executor.h 199 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()
cuda_gpu_executor.cc 833 std::unique_ptr<internal::KernelInterface>
835 return std::unique_ptr<internal::KernelInterface>(new CUDAKernel());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.h 192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.h 192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()

Completed in 301 milliseconds