HomeSort by relevance Sort by last modified time
    Searched refs:cuda_kernel (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 212 CUDAKernel *cuda_kernel = AsCUDAKernel(kernel); local
282 cuda_kernel->cuda_function_ptr())) {
288 cuda_kernel->set_arity(spec.arity());
291 if (!GetKernelMetadata(cuda_kernel, &kernel_metadata)) {
330 bool CUDAExecutor::GetKernelMetadata(CUDAKernel *cuda_kernel,
334 *cuda_kernel->cuda_function_ptr(),
341 *cuda_kernel->cuda_function_ptr(),
355 const CUDAKernel *cuda_kernel = AsCUDAKernel(&kernel); local
356 CUfunction cufunc = cuda_kernel->AsCUDAFunctionValue();
371 if (cuda_kernel->GetPreferredCacheConfig() !
    [all...]
cuda_gpu_executor.h 28 #include "tensorflow/stream_executor/cuda/cuda_kernel.h"
227 bool GetKernelMetadata(CUDAKernel *cuda_kernel,

Completed in 153 milliseconds