Searched refs:src_dev_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 1044 Status s = PrepareRecvTensor(parsed, &src_dev_); in RecvHandler()
1054 checksum_ = Checksum(src_dev_, send_args.device_context, in); in RecvHandler()
1061 if (src_dev_->tensorflow_gpu_device_info() && !on_host) { in RecvHandler()
1065 << "send dev name: " << src_dev_->name() in RecvHandler()
1066 << " gpu_info: " << src_dev_->tensorflow_gpu_device_info(); in RecvHandler()
1079 StreamGPUOp(src_dev_, send_dev_context, in RecvHandler()
1094 src_dev_, send_dev_context, &in, ©, in RecvHandler()
1100 in, src_dev_, send_args.device_context, &proto, is_dead, in RecvHandler()
1151 Allocator* allocator = src_dev_->GetAllocator(host_alloc_attrs); in Clone()
|
| D | rdma.h | 354 Device* src_dev_ = nullptr; variable
|