Searched refs:original_base_shape (Results 1 – 2 of 2) sorted by relevance
343 const Shape& original_base_shape);
1557 const Shape& original_base_shape) { in GetPerGroupBaseShape() argument1558 auto result = original_base_shape; in GetPerGroupBaseShape()1561 if (dim >= original_base_shape.rank()) { in GetPerGroupBaseShape()