Searched refs:shape_pre (Results 1 – 2 of 2) sorted by relevance
1118 ShapeUtil::InsertedOrDeleted1SizedDimensions(const Shape& shape_pre, in InsertedOrDeleted1SizedDimensions() argument1120 CHECK(shape_pre.IsArray()); in InsertedOrDeleted1SizedDimensions()1132 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()1138 if (shape_pre.dimensions(modified_input_dim) > 1) { in InsertedOrDeleted1SizedDimensions()1155 DimensionsUnmodifiedByReshape(shape_pre, shape_post); in InsertedOrDeleted1SizedDimensions()1168 : std::make_pair(shape_pre.rank(), shape_post.rank()); in InsertedOrDeleted1SizedDimensions()
621 InsertedOrDeleted1SizedDimensions(const Shape& shape_pre,