diff options
Diffstat (limited to 'tensorflow/contrib/verbs/verbs_util.cc')
-rw-r--r-- | tensorflow/contrib/verbs/verbs_util.cc | 49 |
1 files changed, 0 insertions, 49 deletions
diff --git a/tensorflow/contrib/verbs/verbs_util.cc b/tensorflow/contrib/verbs/verbs_util.cc index 76e44d34a9..4f5c731a18 100644 --- a/tensorflow/contrib/verbs/verbs_util.cc +++ b/tensorflow/contrib/verbs/verbs_util.cc @@ -20,55 +20,6 @@ limitations under the License. #include "tensorflow/core/lib/strings/str_util.h" namespace tensorflow { -// static sync wrapper: -Status VerbsUtil::CopyGPUTensorToCPUSync(Device* gpu_device, - const DeviceContext* device_context, - const Tensor* gpu_tensor, - Tensor* cpu_tensor) { - Notification n; - Status status; - GPUUtil::CopyGPUTensorToCPU(gpu_device, device_context, - gpu_tensor, cpu_tensor, - [&n, &status](const Status& s) { - status = s; - n.Notify(); - }); - n.WaitForNotification(); - return status; -} - -// static sync wrapper: -Status VerbsUtil::CopyCPUTensorToGPUSync(const Tensor* cpu_tensor, - const DeviceContext* device_context, - Device* gpu_device, - Tensor* gpu_tensor) { - Notification n; - Status status; - GPUUtil::CopyCPUTensorToGPU(cpu_tensor, device_context, - gpu_device, gpu_tensor, - [&n, &status](const Status& s) { - status = s; - n.Notify(); - }); - n.WaitForNotification(); - return status; -} - -// static sync wrapper: -Status VerbsUtil::SetProtoFromGPUSync(const Tensor& tensor, Device* dev, - const DeviceContext* device_context, - TensorProto* proto, bool is_dead) { - Notification n; - Status status; - GPUUtil::SetProtoFromGPU(tensor, dev, device_context, proto, is_dead, - [&n, &status](const Status& s) { - status = s; - n.Notify(); - }); - n.WaitForNotification(); - return status; -} - // static string VerbsUtil::AppendStepidToKey(const string& key, int64 step_id) { return strings::StrCat(key, ";", step_id); |