Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBinaryDivTruncKernel.cu18 void div_trunc_kernel_cuda(TensorIteratorBase& iter) { in div_trunc_kernel_cuda() function
51 REGISTER_DISPATCH(div_trunc_stub, &binary_internal::div_trunc_kernel_cuda);
H A DBinaryInternal.h45 void div_trunc_kernel_cuda(TensorIteratorBase& iter);
H A DBinaryDivFloorKernel.cu28 return div_trunc_kernel_cuda(iter); in div_floor_kernel_cuda()