Home
last modified time | relevance | path

Searched refs:tensorRank (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/detail/
DLazyNVRTC.cpp178 cuuint32_t tensorRank, in cuTensorMapEncodeTiled() argument
196 tensorRank, in cuTensorMapEncodeTiled()
/external/pytorch/aten/src/ATen/native/cuda/
DRowwiseScaledMM.cu20 cuuint32_t tensorRank, in nvrtc_cuTensorMapEncodeTiled() argument
33 tensorRank, in nvrtc_cuTensorMapEncodeTiled()