Searched refs:on_host_shape_ (Results 1 – 6 of 6) sorted by relevance
38 on_host_shape_ = ShapeUtil::DeviceShapeToHostShape(on_device_shape_); in ShapedBuffer()46 : on_host_shape_(std::move(s.on_host_shape_)), in ShapedBuffer()58 on_host_shape_ = std::move(s.on_host_shape_); in operator =()
65 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape()105 on_host_shape_ = ShapeUtil::DeviceShapeToHostShape(on_device_shape); in set_shapes()127 Shape on_host_shape_;
270 on_host_shape_(std::move(on_host_shape)), in PyTpuBuffer()297 host_value->value = std::make_shared<Literal>(on_host_shape_); in CopyToHostAsync()309 if (!on_host_shape_.IsTuple()) { in CopyToHostAsync()368 if (!on_host_shape_.IsTuple()) { in DestructureTuple()372 ShapeUtil::HumanString(on_host_shape_)); in DestructureTuple()379 int num_children = ShapeUtil::TupleElementCount(on_host_shape_); in DestructureTuple()384 on_host_shape_.tuple_shapes(i), child_buffers_.at(i), in DestructureTuple()393 if (on_host_shape_.IsTuple()) { in CopyToDevice()400 on_host_shape_, src_device_buffer, in CopyToDevice()408 on_host_shape_, in CopyToDevice()
208 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape()262 const Shape on_host_shape_; variable
281 const xla::Shape on_host_shape_; variable
153 on_host_shape_(on_host_shape), in XRTTupleAllocation()349 return on_host_shape_; in on_host_shape()