Home
last modified time | relevance | path

Searched defs:tensorRank (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DLazyNVRTC.cpp178 cuuint32_t tensorRank, in cuTensorMapEncodeTiled()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRowwiseScaledMM.cu20 cuuint32_t tensorRank, in nvrtc_cuTensorMapEncodeTiled()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp4913 constexpr size_t tensorRank = sizeof(tensorDimensions) / sizeof(tensorDimensions[0]); in reverseTest() local
4949 constexpr size_t tensorRank = sizeof(tensorDimensions) / sizeof(tensorDimensions[0]); in reverseTestBadQuantization() local
4996 constexpr size_t tensorRank = sizeof(tensorDimensions) / sizeof(tensorDimensions[0]); in reverseTestBadRank() local