Searched refs:shape_post (Results 1 – 2 of 2) sorted by relevance
1197 const Shape& shape_post) { in InsertedOrDeleted1SizedDimensions() argument1199 CHECK(shape_post.IsArray()); in InsertedOrDeleted1SizedDimensions()1208 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()1222 if (shape_post.dimensions(modified_output_dim) > 1) { in InsertedOrDeleted1SizedDimensions()1231 DimensionsUnmodifiedByReshape(shape_pre, shape_post); in InsertedOrDeleted1SizedDimensions()1244 : std::make_pair(shape_pre.rank(), shape_post.rank()); in InsertedOrDeleted1SizedDimensions()
531 const Shape& shape_post);