Searched refs:lhs_reduction_dimension (Results 1 – 1 of 1) sorted by relevance
559 int64 lhs_reduction_dimension = dim_nums.lhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm() local563 CHECK_EQ(lhs_shape.dimensions(lhs_reduction_dimension), in EmitNaiveLlvmIrGemm()567 lhs_reduction_dimension == LayoutUtil::Minor(lhs_shape.layout(), 0); in EmitNaiveLlvmIrGemm()577 lhs_array_, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in EmitNaiveLlvmIrGemm()591 0, lhs_shape.dimensions(lhs_reduction_dimension), "reduction", in EmitNaiveLlvmIrGemm()600 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()676 if (dimension != lhs_reduction_dimension) { in EmitNaiveLlvmIrGemm()