Home
last modified time | relevance | path

Searched refs:CreateKernelImplementation (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc72 implementation_(parent->implementation()->CreateKernelImplementation()) {} in KernelBase()
Dstream_executor_internal.h359 virtual std::unique_ptr<KernelInterface> CreateKernelImplementation() = 0;
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h192 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation() function
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h227 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc847 GpuExecutor::CreateKernelImplementation() { in CreateKernelImplementation() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc945 GpuExecutor::CreateKernelImplementation() { in CreateKernelImplementation() function in stream_executor::gpu::GpuExecutor