Searched refs:new_shape_dims (Results 1 – 1 of 1) sorted by relevance
449 std::vector<int64> new_shape_dims; in CollapseFirstNDims() local450 new_shape_dims.reserve(operand_shape.dimensions_size() - n + 1); in CollapseFirstNDims()451 new_shape_dims.push_back(new_shape_leading_bound); in CollapseFirstNDims()455 std::back_inserter(new_shape_dims)); in CollapseFirstNDims()458 ShapeUtil::MakeShape(operand_shape.element_type(), new_shape_dims); in CollapseFirstNDims()466 std::vector<int64> new_shape_dims; in PrependDegenerateDims() local468 new_shape_dims.reserve(n + operand_shape.dimensions_size()); in PrependDegenerateDims()469 new_shape_dims.insert(new_shape_dims.begin(), n, 1); in PrependDegenerateDims()470 absl::c_copy(operand_shape.dimensions(), std::back_inserter(new_shape_dims)); in PrependDegenerateDims()471 return MakeReshapeHlo(new_shape_dims, operand); in PrependDegenerateDims()