Searched refs:DimensionsUnmodifiedByReshape (Results 1 – 5 of 5) sorted by relevance
634 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()644 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()656 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()
1026 DimensionsUnmodifiedByReshape(shape_pre, shape_post); in InsertedOrDeleted1SizedDimensions()1049 ShapeUtil::DimensionsUnmodifiedByReshape(const Shape& input_shape, in DimensionsUnmodifiedByReshape() function in xla::ShapeUtil1079 ShapeUtil::DimensionsUnmodifiedByReshape(from_shape, to_shape); in ReshapeLeavesDimensionsUnmodified()
591 static std::vector<std::pair<int64, int64>> DimensionsUnmodifiedByReshape(
1684 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims()3662 ShapeUtil::DimensionsUnmodifiedByReshape(arg->operand(0)->shape(), in HandleReduce()
521 ShapeUtil::DimensionsUnmodifiedByReshape(*operand_shape, reshaped_shape); in AddBroadcastSequence()