Searched defs:batch_ptr_s (Results 1 – 1 of 1) sorted by relevance
208 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in COOMatrixToSparseTensorKernel3D()274 GpuDeviceArrayStruct<int> batch_ptr_s, const int batch_size, in CSRSparseMatrixBatchMulVecKernel3D()409 const int size, const int rows, GpuDeviceArrayStruct<int> batch_ptr_s, in CSRSparseMatrixSoftmaxKernel3D()