Searched defs:host_batch_ptr (Results 1 – 1 of 1) sorted by relevance
232 TTypes<int>::ConstVec host_batch_ptr, TTypes<int>::Vec coo_row_ind, in operator ()()314 auto host_batch_ptr = a.batch_pointers().vec<int>(); in CSRSparseMatrixBatchMulVecImpl() local436 auto host_batch_ptr = logits.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl() local