Home
last modified time | relevance | path

Searched defs:nnzTotalDevHostPtr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Drocm_sparse.cc300 int* csrSortedRowPtrC, int* nnzTotalDevHostPtr) { in CsrgemmNnz()
471 int* csrSortedRowPtrC, int* nnzTotalDevHostPtr, void* workspace) { in CsrgeamNnz()
H A Dcuda_sparse.cc372 int* csrSortedRowPtrC, int* nnzTotalDevHostPtr, void* workspace) { in CsrgeamNnz()
769 int* nnzTotalDevHostPtr) { in CsrgemmNnz()
857 int* csrSortedRowPtrC, int* nnzTotalDevHostPtr, csrgemm2Info_t info, in CsrgemmNnz()