Searched refs:DimensionsUnmodifiedByReshape (Results 1 – 6 of 6) sorted by relevance
612 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()622 EXPECT_THAT(ShapeUtil::DimensionsUnmodifiedByReshape( in TEST()634 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::ShapeUtil
578 static std::vector<std::pair<int64, int64>> DimensionsUnmodifiedByReshape(
336 ShapeUtil::DimensionsUnmodifiedByReshape(operand->shape(), in HandleReshape()
1974 ShapeUtil::DimensionsUnmodifiedByReshape(hlo->operand(0)->shape(), in ReshapeLeavesDimensionsUnmodified()3020 ShapeUtil::DimensionsUnmodifiedByReshape(arg->operand(0)->shape(), in HandleReduce()
2708 ShapeUtil::DimensionsUnmodifiedByReshape(operand, inferred_shape); in InferReshapeShape()