Searched refs:new_source_shape_dims (Results 1 – 1 of 1) sorted by relevance
483 DimensionVector new_source_shape_dims; in ReshapeToRemoveDegenerateDims() local486 new_source_shape_dims.push_back(source_shape.dimensions(i)); in ReshapeToRemoveDegenerateDims()491 ShapeUtil::MakeShape(shape.element_type(), new_source_shape_dims); in ReshapeToRemoveDegenerateDims()569 DimensionVector new_source_shape_dims = new_result_shape_dims; in ReshapeToAddDegenerateDims() local571 EraseAt(&new_source_shape_dims, output_dim); in ReshapeToAddDegenerateDims()575 for (int i = 0, e = new_source_shape_dims.size(); i < e; i++) { in ReshapeToAddDegenerateDims()580 if (new_source_shape_dims[new_source_dim] != 1) { in ReshapeToAddDegenerateDims()589 InsertAt(&new_source_shape_dims, /*index=*/new_source_dim, in ReshapeToAddDegenerateDims()593 new_source_shape_dims); in ReshapeToAddDegenerateDims()