Searched defs:rhs_shmem (Results 1 – 1 of 1) sorted by relevance
23 const OutputMapper output, Scalar* lhs_shmem, Scalar* rhs_shmem, in EigenContractionKernelInternal()515 __shared__ Scalar rhs_shmem[72 * 64]; in EigenContractionKernel() local1147 __shared__ float2 rhs_shmem[128*8]; in EigenFloatContractionKernel() local1194 __shared__ float2 rhs_shmem[64][8]; in EigenFloatContractionKernel16x16() local