Searched refs:DimensionsUnmodifiedByReshape (Results 1 – 6 of 6) sorted by relevance
38 ShapeUtil::DimensionsUnmodifiedByReshape(in_shape, out_shape).size() == in IsAddingOnlyDegenerateDimensions()
636 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()646 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()658 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()
1155 DimensionsUnmodifiedByReshape(shape_pre, shape_post); in InsertedOrDeleted1SizedDimensions()1178 ShapeUtil::DimensionsUnmodifiedByReshape(const Shape& input_shape, in DimensionsUnmodifiedByReshape() function in xla::ShapeUtil1208 ShapeUtil::DimensionsUnmodifiedByReshape(from_shape, to_shape); in ReshapeLeavesDimensionsUnmodified()
640 static std::vector<std::pair<int64, int64>> DimensionsUnmodifiedByReshape(
2140 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims()4648 ShapeUtil::DimensionsUnmodifiedByReshape(arg->operand(0)->shape(), in HandleReduce()
599 ShapeUtil::DimensionsUnmodifiedByReshape(*operand_shape, reshaped_shape); in AddBroadcastSequence()