Searched refs:batch_ptr_s (Results 1 – 1 of 1) sorted by relevance
208 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in COOMatrixToSparseTensorKernel3D() argument211 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in COOMatrixToSparseTensorKernel3D()274 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in CSRSparseMatrixBatchMulVecKernel3D() argument278 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in CSRSparseMatrixBatchMulVecKernel3D()409 const int size, const int rows, GpuDeviceArrayStruct<int> batch_ptr_s, in CSRSparseMatrixSoftmaxKernel3D() argument417 CopyFromGpuDeviceArrayToLocal(std::move(batch_ptr_s), local_batch_ptr, in CSRSparseMatrixSoftmaxKernel3D()