Searched defs:lhs_shmem (Results 1 – 1 of 1) sorted by relevance
23 const OutputMapper output, Scalar* lhs_shmem, Scalar* rhs_shmem, in EigenContractionKernelInternal()506 __shared__ Scalar lhs_shmem[72 * 64]; in EigenContractionKernel() local1134 __shared__ float2 lhs_shmem[64*32]; in EigenFloatContractionKernel() local1180 __shared__ float2 lhs_shmem[32][16]; in EigenFloatContractionKernel16x16() local