Searched refs:outer_bs_a (Results 1 – 1 of 1) sorted by relevance
129 void MacroKernel(const char* __restrict a, int64_t lda, int outer_bs_a, in MacroKernel() argument133 << " outer_bs_a=" << outer_bs_a << " outer_bs_b=" << outer_bs_b in MacroKernel()139 DCHECK_EQ(outer_bs_a * inner_bs % 2, 0); in MacroKernel()143 p + outer_bs_a * inner_bs * i, in MacroKernel()144 outer_bs_a * inner_bs / 2); in MacroKernel()147 lda = outer_bs_a * inner_bs * sizeof(float); in MacroKernel()150 for (int i = 0; i < outer_bs_a; ++i) { in MacroKernel()163 void Transpose(const char* __restrict a, int outer_bs_a, char* __restrict b, in Transpose() argument166 DVLOG(10) << "Transpose " << outer_bs_a << " " << outer_bs_b; in Transpose()167 DCHECK_GT(outer_bs_a, 0); in Transpose()[all …]