Searched refs:reshape_back_dims (Results 1 – 1 of 1) sorted by relevance
606 std::vector<int64> reshape_back_dims( in IncreaseSpatialSizeOnSpaceToBatchedShape() local610 reshape_back_dims[spatial_dimension] = new_spatial_dim_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()611 reshape_back_dims[batch_dimension] = new_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()614 MakeReshapeHlo(reshape_back_dims, reshaped_activations)); in IncreaseSpatialSizeOnSpaceToBatchedShape()