Searched defs:pBufferSizeInBytes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDABlas.cpp | 478 … m, int64_t n, int64_t nnz, const int *csrRowPtr, const int *csrColInd, size_t *pBufferSizeInBytes) in Xcsrsort_bufferSizeExt()
507 …64_t m, int64_t n, int64_t nnz, const int *cooRows, const int *cooCols, size_t *pBufferSizeInBytes) in Xcoosort_bufferSizeExt()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | rocm_sparse.cc | 362 size_t pBufferSizeInBytes = 0; in Csru2csrImpl() local
|
| H A D | cuda_sparse.cc | 925 size_t pBufferSizeInBytes = 0; in Csru2csrImpl() local
|