Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/
Djiterator.h15 TORCH_CUDA_CPP_API c10::SmallVector<at::Tensor> CompileAndLaunchKernel(
29 TORCH_CUDA_CPP_API c10::SmallVector<at::Tensor> CompileAndLaunchKernel( in CompileAndLaunchKernel() function
Djiterator.cu321 c10::SmallVector<at::Tensor> CompileAndLaunchKernel( in CompileAndLaunchKernel() function
/external/pytorch/torch/csrc/cuda/
DModule.cpp397 c10::SmallVector<at::Tensor> outputs = at::cuda::CompileAndLaunchKernel( in THCPModule_cudaJiteratorCompileAndLaunchKernel()