Searched refs:new_source_shape_dims (Results 1 – 1 of 1) sorted by relevance
477 DimensionVector new_source_shape_dims; in ReshapeToRemoveDegenerateDims() local480 new_source_shape_dims.push_back(source_shape.dimensions(i)); in ReshapeToRemoveDegenerateDims()485 ShapeUtil::MakeShape(shape.element_type(), new_source_shape_dims); in ReshapeToRemoveDegenerateDims()563 DimensionVector new_source_shape_dims = new_result_shape_dims; in ReshapeToAddDegenerateDims() local565 EraseAt(&new_source_shape_dims, output_dim); in ReshapeToAddDegenerateDims()569 for (int i = 0, e = new_source_shape_dims.size(); i < e; i++) { in ReshapeToAddDegenerateDims()574 if (new_source_shape_dims[new_source_dim] != 1) { in ReshapeToAddDegenerateDims()583 InsertAt(&new_source_shape_dims, /*index=*/new_source_dim, in ReshapeToAddDegenerateDims()587 new_source_shape_dims); in ReshapeToAddDegenerateDims()