Searched defs:on_device_shape (Results 1 – 6 of 6) sorted by relevance
35 const Shape& on_device_shape, in ShapedBuffer()107 const Shape& on_device_shape, in ScopedShapedBuffer()
66 const Shape& on_device_shape() const { return on_device_shape_; } in on_device_shape() function
158 const Shape on_device_shape = HostShapeToDeviceShape(literal.shape()); in TransferArrayToDeviceAsync() local317 const Shape on_device_shape = HostShapeToDeviceShape(on_host_shape); in AllocateScopedShapedBuffer() local
50 xla::Shape on_device_shape = in AllocateShapedBuffer() local
238 const xla::Shape on_device_shape = in PopulateInputs() local
101 xla::Shape on_device_shape = transfer_manager->HostShapeToDeviceShape(shape); in AllocateScopedShapedBuffer() local170 const xla::Shape& on_device_shape) in XRTTupleAllocation()262 const xla::Shape& XRTTupleAllocation::on_device_shape() { in on_device_shape() function in tensorflow::XRTTupleAllocation