Searched refs:concat_shape_dims (Results 1 – 1 of 1) sorted by relevance
197 std::vector<int64> concat_shape_dims; in SplitShapeIntoHalves() local204 concat_shape_dims.push_back(half_shape_dims[i]); in SplitShapeIntoHalves()205 concat_shape_dims.push_back(2); in SplitShapeIntoHalves()208 concat_shape_dims.push_back(shape.dimensions(i)); in SplitShapeIntoHalves()214 ShapeUtil::MakeShape(shape.element_type(), concat_shape_dims); in SplitShapeIntoHalves()