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