Searched refs:remote_mr (Results 1 – 1 of 1) sorted by relevance
398 RemoteMemoryRegion remote_mr; in TransportOptionsFromTensor() local399 remote_mr.set_host(host_); in TransportOptionsFromTensor()400 remote_mr.set_port(port_); in TransportOptionsFromTensor()401 remote_mr.set_addr(reinterpret_cast<uint64_t>(buffer->data())); in TransportOptionsFromTensor()402 remote_mr.set_rkey(mr->rkey); in TransportOptionsFromTensor()403 remote_mr.set_tensor_key(tensor_key); in TransportOptionsFromTensor()404 mutable_transport_options->PackFrom(remote_mr); in TransportOptionsFromTensor()427 RemoteMemoryRegion remote_mr; in TensorFromTransportOptions() local428 if (!transport_options.UnpackTo(&remote_mr)) { in TensorFromTransportOptions()439 std::make_pair(std::make_pair(remote_mr.host(), remote_mr.port()), in TensorFromTransportOptions()[all …]