Searched refs:on_device_shape_ (Results 1 – 4 of 4) sorted by relevance
38 on_device_shape_(on_device_shape), in ShapedBuffer()41 buffers_(&on_device_shape_) {} in ShapedBuffer()45 on_device_shape_(std::move(s.on_device_shape_)), in ShapedBuffer()52 buffers_.replace_shape_ptr(&on_device_shape_); in ShapedBuffer()57 on_device_shape_ = std::move(s.on_device_shape_); in operator =()64 buffers_.replace_shape_ptr(&on_device_shape_); in operator =()
66 const Shape& on_device_shape() const { return on_device_shape_; } in on_device_shape()91 CHECK(ShapeUtil::Equal(buffers.shape(), on_device_shape_)); in set_buffers()110 Shape on_device_shape_; variable
174 on_device_shape_(on_device_shape), in XRTTupleAllocation()175 buffers_(&on_device_shape_) {} in XRTTupleAllocation()263 return on_device_shape_; in on_device_shape()
224 const xla::Shape on_device_shape_; variable