Searched refs:dims_to_insert (Results 1 – 2 of 2) sorted by relevance
389 HloInstruction* operand, absl::Span<const int64> dims_to_insert) { in InsertDegenerateDims() argument390 CHECK(absl::c_is_sorted(dims_to_insert)); in InsertDegenerateDims()394 operand_shape.dimensions_size() + dims_to_insert.size(); in InsertDegenerateDims()395 for (auto dim_to_insert : dims_to_insert) { in InsertDegenerateDims()404 if (dims_to_insert_idx < dims_to_insert.size() && in InsertDegenerateDims()405 i == dims_to_insert[dims_to_insert_idx]) { in InsertDegenerateDims()
205 HloInstruction* operand, absl::Span<const int64> dims_to_insert);