Searched refs:lhs_shmem (Results 1 – 1 of 1) sorted by relevance
23 const OutputMapper output, Scalar* lhs_shmem, Scalar* rhs_shmem, in EigenContractionKernelInternal() argument237 lhs_shmem[lhs_store_idx_0] = lhs_pf0; \ in EigenContractionKernelInternal()240 lhs_shmem[lhs_store_idx_1] = lhs_pf1; \ in EigenContractionKernelInternal()243 lhs_shmem[lhs_store_idx_2] = lhs_pf2; \ in EigenContractionKernelInternal()246 lhs_shmem[lhs_store_idx_3] = lhs_pf3; \ in EigenContractionKernelInternal()249 lhs_shmem[lhs_store_idx_4] = lhs_pf4; \ in EigenContractionKernelInternal()252 lhs_shmem[lhs_store_idx_5] = lhs_pf5; \ in EigenContractionKernelInternal()255 lhs_shmem[lhs_store_idx_6] = lhs_pf6; \ in EigenContractionKernelInternal()258 lhs_shmem[lhs_store_idx_7] = lhs_pf7; \ in EigenContractionKernelInternal()322 const Scalar* lhs_block = &lhs_shmem[threadIdx.x + 9 * threadIdx.y]; in EigenContractionKernelInternal()[all …]