Searched refs:rhs_reduction_dimension (Results 1 – 1 of 1) sorted by relevance
624 int64_t rhs_reduction_dimension = dim_nums.rhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm() local628 rhs_shape.dimensions(rhs_reduction_dimension)); in EmitNaiveLlvmIrGemm()633 rhs_reduction_dimension == LayoutUtil::Minor(rhs_shape.layout(), 0); in EmitNaiveLlvmIrGemm()644 rhs_array_, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in EmitNaiveLlvmIrGemm()667 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()745 if (dimension != rhs_reduction_dimension) { in EmitNaiveLlvmIrGemm()