Searched refs:cuda_kernel (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 269 GpuKernel* cuda_kernel = AsGpuKernel(kernel); in GetKernel() local
310 cuda_kernel->gpu_function_ptr())) { in GetKernel()
316 cuda_kernel->set_arity(spec.arity()); in GetKernel()
319 if (!GetKernelMetadata(cuda_kernel, &kernel_metadata)) { in GetKernel()
402 bool GpuExecutor::GetKernelMetadata(GpuKernel* cuda_kernel, in GetKernelMetadata() argument
406 *cuda_kernel->gpu_function_ptr(), &value)) { in GetKernelMetadata()
412 *cuda_kernel->gpu_function_ptr(), &value)) { in GetKernelMetadata()
425 const GpuKernel* cuda_kernel = AsGpuKernel(&kernel); in Launch() local
426 CUfunction cufunc = cuda_kernel->AsGpuFunctionHandle(); in Launch()
441 if (cuda_kernel->GetPreferredCacheConfig() != in Launch()
[all …]
|
| D | BUILD | 169 ":cuda_kernel",
341 name = "cuda_kernel",
342 srcs = if_cuda_is_configured(["cuda_kernel.cc"]),
343 hdrs = if_cuda_is_configured(["cuda_kernel.h"]),
417 ":cuda_kernel",
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 264 bool GetKernelMetadata(GpuKernel* cuda_kernel,
|