Searched refs:rhs_reduction_dimension (Results 1 – 2 of 2) sorted by relevance
511 const int64 rhs_reduction_dimension = dnums.rhs_contracting_dimensions(0); in HandleDot() local516 CHECK_NE(rhs_reduction_dimension, batch_dim); in HandleDot()521 rhs_shape.dimensions(rhs_reduction_dimension)) in HandleDot()524 << ", and rhs_shape.dimensions(" << rhs_reduction_dimension in HandleDot()525 << ") = " << rhs_shape.dimensions(rhs_reduction_dimension); in HandleDot()536 rhs_array, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in HandleDot()554 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()618 if (dimension != rhs_reduction_dimension) { in HandleDot()
417 int64 rhs_reduction_dimension = dim_nums.rhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm() local421 rhs_shape.dimensions(rhs_reduction_dimension)); in EmitNaiveLlvmIrGemm()426 rhs_reduction_dimension == LayoutUtil::Minor(rhs_shape.layout(), 0); in EmitNaiveLlvmIrGemm()437 rhs_array_, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in EmitNaiveLlvmIrGemm()460 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()532 if (dimension != rhs_reduction_dimension) { in EmitNaiveLlvmIrGemm()