Searched refs:new_shape_dims (Results 1 – 1 of 1) sorted by relevance
321 std::vector<int64> new_shape_dims; in CollapseFirstNDims() local322 new_shape_dims.reserve(operand_shape.dimensions_size() - n + 1); in CollapseFirstNDims()323 new_shape_dims.push_back(new_shape_leading_bound); in CollapseFirstNDims()327 std::back_inserter(new_shape_dims)); in CollapseFirstNDims()330 ShapeUtil::MakeShape(operand_shape.element_type(), new_shape_dims); in CollapseFirstNDims()338 std::vector<int64> new_shape_dims; in PrependDegenerateDims() local340 new_shape_dims.reserve(n + operand_shape.dimensions_size()); in PrependDegenerateDims()341 new_shape_dims.insert(new_shape_dims.begin(), n, 1); in PrependDegenerateDims()342 absl::c_copy(operand_shape.dimensions(), std::back_inserter(new_shape_dims)); in PrependDegenerateDims()343 return MakeReshapeHlo(new_shape_dims, operand); in PrependDegenerateDims()