Method replace_shape_ptr

tensorflow/compiler/xla/shape_tree.h:130–136

A ShapeTree object can own the underlying Shape pointer (via the shape_storage_ member), or can point to a Shape object owned by the caller. This API replaces the underlying Shape object to the one supplied by the caller, whom must ensure the object remain valid for the whole lifetime of this ShapeTree object, and also that the Shape is consistent with it.

  // caller, whom must ensure the object remain valid for the whole lifetime of
  // this ShapeTree object, and also that the Shape is consistent with it.
  void replace_shape_ptr(const Shape* shape) {
    if (shape_storage_ != nullptr) {
      DCHECK_EQ(*shape, *shape_storage_);
      shape_storage_ = nullptr;
    }
    shape_ = shape;
  }

  // Returns true if the node at the given index is a leaf node (an array
  // shape).