Searched refs:dims_to_elide (Results 1 – 2 of 2) sorted by relevance
364 HloInstruction* operand, absl::Span<const int64> dims_to_elide) { in ElideDegenerateDims() argument365 CHECK(absl::c_is_sorted(dims_to_elide)); in ElideDegenerateDims()371 dims_to_elide.size()); in ElideDegenerateDims()372 int64 dims_to_elide_idx = dims_to_elide.size() - 1; in ElideDegenerateDims()374 if (dims_to_elide_idx >= 0 && i == dims_to_elide[dims_to_elide_idx]) { in ElideDegenerateDims()
195 HloInstruction* operand, absl::Span<const int64> dims_to_elide);