Searched refs:dst_dev_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 1484 dst_dev_(dst_dev), in RdmaTensorRequest()
1506 : Checksum(dst_dev_, recv_args_.device_context, val); in Done()
1531 new Tensor(dst_dev_->GetAllocator(recv_args_.alloc_attrs), in AllocateTensors()
1567 if (dst_dev_->tensorflow_gpu_device_info() && !on_host && in AllocateTensorsAsync()
1571 StreamGPUOp(dst_dev_, recv_args_.device_context, done); in AllocateTensorsAsync()
1633 dst_dev_, result_tensor_, in RecvTensorContent()
1652 Status s = dst_dev_->MakeTensorFromProto(proto, recv_args_.alloc_attrs, in RecvTensorContent()
|
| D | rdma.h | 293 Device* dst_dev_; variable
|