Searched refs:cuda_kernel (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 265 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel() local
300 cuda_kernel->gpu_function_ptr())) { in GetKernel()
306 cuda_kernel->set_arity(spec.arity()); in GetKernel()
309 TF_RETURN_IF_ERROR(GetKernelMetadata(cuda_kernel, &kernel_metadata)); in GetKernel()
386 port::Status GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata() argument
390 CU_FUNC_ATTRIBUTE_NUM_REGS, *cuda_kernel->gpu_function_ptr(), &value)); in GetKernelMetadata()
395 *cuda_kernel->gpu_function_ptr(), &value)); in GetKernelMetadata()
406 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch() local
407 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in Launch()
422 if (cuda_kernel->GetPreferredCacheConfig() != in Launch()
[all …]
|
| D | BUILD | 227 ":cuda_kernel",
534 name = "cuda_kernel",
535 srcs = if_cuda_is_configured(["cuda_kernel.cc"]),
536 hdrs = if_cuda_is_configured(["cuda_kernel.h"]),
620 ":cuda_kernel",
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 262 port::Status GetKernelMetadata(GpuKernel* cuda_kernel,
|