Searched refs:dims_to_insert (Results 1 – 2 of 2) sorted by relevance
620 HloInstruction* operand, absl::Span<const int64_t> dims_to_insert) { in InsertDegenerateDims() argument621 CHECK(absl::c_is_sorted(dims_to_insert)); in InsertDegenerateDims()625 operand_shape.dimensions_size() + dims_to_insert.size(); in InsertDegenerateDims()626 for (auto dim_to_insert : dims_to_insert) { in InsertDegenerateDims()635 if (dims_to_insert_idx < dims_to_insert.size() && in InsertDegenerateDims()636 i == dims_to_insert[dims_to_insert_idx]) { in InsertDegenerateDims()
336 HloInstruction* operand, absl::Span<const int64_t> dims_to_insert);