Searched refs:min_array_rank_ (Results 1 – 2 of 2) sorted by relevance
35 min_array_rank_(min_array_rank) {} in decompose_to_tuple_()44 int64 min_array_rank_; variable
53 return all_to_all->shape().rank() < min_array_rank_; in InstructionMatchesPattern()76 for (int64 j = all_to_all->shape().rank() + 1; j < min_array_rank_; ++j) { in ExpandInstruction()81 LayoutUtil::GetDefaultLayoutForRank(min_array_rank_); in ExpandInstruction()