| 20 | namespace tensorflow { |
| 21 | |
| 22 | TensorShape ReductionHelper::out_reshape() const { |
| 23 | TensorShape shape; |
| 24 | for (auto size : out_reshape_) shape.AddDim(size); |
| 25 | return shape; |
| 26 | } |
| 27 | |
| 28 | // The final output shape must be allocated with this shape. |
| 29 | TensorShape ReductionHelper::out_shape() const { |