Searched defs:workspaceInBytesOnDevice (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | CUDASolver.cpp | 1044 …int64_t lda, cudaDataType computeType, size_t *workspaceInBytesOnDevice, size_t *workspaceInBytesO… in xpotrf_buffersize()
1052 …int64_t lda, cudaDataType computeType, void *bufferOnDevice, size_t workspaceInBytesOnDevice, void… in xpotrf()
1720 size_t* workspaceInBytesOnDevice, in xsyevd_bufferSize()
1748 size_t* workspaceInBytesOnDevice, in xsyevd_bufferSize()
1776 size_t* workspaceInBytesOnDevice, in xsyevd_bufferSize()
1804 size_t* workspaceInBytesOnDevice, in xsyevd_bufferSize()
1833 size_t workspaceInBytesOnDevice, in xsyevd()
1867 size_t workspaceInBytesOnDevice, in xsyevd()
1901 size_t workspaceInBytesOnDevice, in xsyevd()
1935 size_t workspaceInBytesOnDevice, in xsyevd()
|