Searched refs:mat_mult_dims (Results 1 – 1 of 1) sorted by relevance
227 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemm() local232 int64 m = mat_mult_dims.m; in EmitTiledLlvmIrGemm()233 int64 k = mat_mult_dims.k; in EmitTiledLlvmIrGemm()234 int64 n = mat_mult_dims.n; in EmitTiledLlvmIrGemm()236 if (mat_mult_dims.lhs_column_major) { in EmitTiledLlvmIrGemm()269 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemv() local276 if (mat_mult_dims.m == 1) { in EmitTiledLlvmIrGemv()278 mat_mult_dims.rhs_non_canonical ^ !mat_mult_dims.rhs_column_major; in EmitTiledLlvmIrGemv()280 k = mat_mult_dims.k; in EmitTiledLlvmIrGemv()281 m = mat_mult_dims.n; in EmitTiledLlvmIrGemv()[all …]