Searched refs: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 ()() argument253 GpuDeviceArrayOnHost<int> batch_ptr_copy(c, host_batch_ptr.size()); in operator ()()256 batch_ptr_copy.Set(i, host_batch_ptr(i)); in operator ()()314 auto host_batch_ptr = a.batch_pointers().vec<int>(); in CSRSparseMatrixBatchMulVecImpl() local321 GpuDeviceArrayOnHost<int> batch_ptr_copy(ctx, host_batch_ptr.size()); in CSRSparseMatrixBatchMulVecImpl()324 batch_ptr_copy.Set(i, host_batch_ptr(i)); in CSRSparseMatrixBatchMulVecImpl()436 auto host_batch_ptr = logits.batch_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGPUImpl() local454 DCHECK_EQ(batch_size, host_batch_ptr.size() - 1); in CSRSparseMatrixSoftmaxGPUImpl()458 GpuDeviceArrayOnHost<int> batch_ptr_copy(ctx, host_batch_ptr.size()); in CSRSparseMatrixSoftmaxGPUImpl()460 for (int i = 0; i < host_batch_ptr.size(); ++i) { in CSRSparseMatrixSoftmaxGPUImpl()[all …]