Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc369 std::vector<int64> new_shape_dim_bounds; in ElideDegenerateDims() local
370 new_shape_dim_bounds.reserve(input_shape.dimensions_size() - in ElideDegenerateDims()
378 new_shape_dim_bounds.push_back(input_shape.dimensions(i)); in ElideDegenerateDims()
382 absl::c_reverse(new_shape_dim_bounds); in ElideDegenerateDims()
384 ShapeUtil::MakeShape(input_shape.element_type(), new_shape_dim_bounds); in ElideDegenerateDims()