Method out_shape

tensorflow/core/kernels/reduction_ops_common.cc:29–33

The final output shape must be allocated with this shape.

// The final output shape must be allocated with this shape.
TensorShape ReductionHelper::out_shape() const {
  TensorShape shape;
  for (auto size : out_shape_) shape.AddDim(size);
  return shape;
}

TensorShape ReductionHelper::shuffled_shape() {
  const int dims = data_reshape_.size();