Searched refs:mat_mult_dims (Results 1 – 1 of 1) sorted by relevance
381 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemm() local386 int64_t m = mat_mult_dims.m; in EmitTiledLlvmIrGemm()387 int64_t k = mat_mult_dims.k; in EmitTiledLlvmIrGemm()388 int64_t n = mat_mult_dims.n; in EmitTiledLlvmIrGemm()390 if (mat_mult_dims.lhs_column_major) { in EmitTiledLlvmIrGemm()424 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemv() local431 if (mat_mult_dims.m == 1) { in EmitTiledLlvmIrGemv()443 bool rhs_effectively_column_major = mat_mult_dims.rhs_canonical in EmitTiledLlvmIrGemv()444 ? mat_mult_dims.rhs_column_major in EmitTiledLlvmIrGemv()445 : !mat_mult_dims.rhs_column_major; in EmitTiledLlvmIrGemv()[all …]