Home
last modified time | relevance | path

Searched refs:GetKernelMetadata (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h262 port::Status GetKernelMetadata(GpuKernel* cuda_kernel,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc279 TF_RETURN_IF_ERROR(GetKernelMetadata(rocm_kernel, &kernel_metadata)); in GetKernel()
285 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* rocm_kernel, in GetKernelMetadata() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc309 TF_RETURN_IF_ERROR(GetKernelMetadata(cuda_kernel, &kernel_metadata)); in GetKernel()
386 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata() function in stream_executor::gpu::GpuExecutor