Searched refs:rhs_reduction_dimension (Results 1 – 1 of 1) sorted by relevance
560 int64 rhs_reduction_dimension = dim_nums.rhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm() local564 rhs_shape.dimensions(rhs_reduction_dimension)); in EmitNaiveLlvmIrGemm()569 rhs_reduction_dimension == LayoutUtil::Minor(rhs_shape.layout(), 0); in EmitNaiveLlvmIrGemm()580 rhs_array_, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in EmitNaiveLlvmIrGemm()603 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()681 if (dimension != rhs_reduction_dimension) { in EmitNaiveLlvmIrGemm()