Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc242 GpuKernel* rocm_kernel = AsGpuKernel(kernel); in GetKernel() local
287 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* rocm_kernel, in GetKernelMetadata()
304 const GpuKernel* rocm_kernel = AsGpuKernel(&kernel); in Launch() local