// backward information flow is added in the future.
  return StoreOutputShapes(*graph, shape_refiner, shape_info);
}

absl::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a,
                                                  const InferredShape& b) {
  InferredShape result;
  TF_RETURN_IF_ERROR(a.shape.MergeWith(b.shape, &result.shape));