Home
last modified time | relevance | path

Searched defs:CreateKernelImplementation (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.h166 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor.h182 CreateKernelImplementation() override { in CreateKernelImplementation() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.h188 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc789 GpuExecutor::CreateKernelImplementation() { in CreateKernelImplementation() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc631 std::unique_ptr<internal::KernelInterface> CreateKernelImplementation() in CreateKernelImplementation() function in stream_executor::__anon702966b10111::CStreamExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc918 GpuExecutor::CreateKernelImplementation() { in CreateKernelImplementation() function in stream_executor::gpu::GpuExecutor