Home
last modified time | relevance | path

Searched refs:new_shape_dims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc321 std::vector<int64> new_shape_dims; in CollapseFirstNDims() local
322 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() local
340 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()