Searched refs:on_host_shape_ (Results 1 – 4 of 4) sorted by relevance
37 : on_host_shape_(on_host_shape), in ShapedBuffer()44 : on_host_shape_(std::move(s.on_host_shape_)), in ShapedBuffer()56 on_host_shape_ = std::move(s.on_host_shape_); in operator =()
62 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape()107 Shape on_host_shape_;
221 const xla::Shape on_host_shape_; variable
173 on_host_shape_(on_host_shape), in XRTTupleAllocation()260 const xla::Shape& XRTTupleAllocation::on_host_shape() { return on_host_shape_; } in on_host_shape()