Searched refs:half_shape_dims (Results 1 – 1 of 1) sorted by relevance
196 std::vector<int64> half_shape_dims; in SplitShapeIntoHalves() local202 half_shape_dims.push_back(CeilOfRatio<int64>(shape.dimensions(i), 2)); in SplitShapeIntoHalves()203 half_shape_dims.push_back(1); in SplitShapeIntoHalves()204 concat_shape_dims.push_back(half_shape_dims[i]); in SplitShapeIntoHalves()207 half_shape_dims.push_back(shape.dimensions(i)); in SplitShapeIntoHalves()212 pair.half_shape = ShapeUtil::MakeShape(shape.element_type(), half_shape_dims); in SplitShapeIntoHalves()