Searched defs:DevicePutResult (Results 1 – 1 of 1) sorted by relevance
141 struct DevicePutResult { struct142 explicit DevicePutResult(xla::PjRtBuffer* b, bool weak_type) in DevicePutResult() argument144 DevicePutResult(std::unique_ptr<xla::PjRtBuffer> new_buffer, bool weak_type) in DevicePutResult() argument168 xla::StatusOr<DevicePutResult> DevicePut(pybind11::handle arg, argument