Home
last modified time | relevance | path

Searched refs:shape_pre (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc998 ShapeUtil::InsertedOrDeleted1SizedDimensions(const Shape& shape_pre, in InsertedOrDeleted1SizedDimensions() argument
1009 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()
1015 if (shape_pre.dimensions(modified_input_dim) > 1) { in InsertedOrDeleted1SizedDimensions()
1032 DimensionsUnmodifiedByReshape(shape_pre, shape_post); in InsertedOrDeleted1SizedDimensions()
1045 : std::make_pair(Rank(shape_pre), Rank(shape_post)); in InsertedOrDeleted1SizedDimensions()
Dshape_util.h500 InsertedOrDeleted1SizedDimensions(const Shape& shape_pre,