diff options
Diffstat (limited to 'tensorflow/contrib/verbs/verbs_util.h')
-rw-r--r-- | tensorflow/contrib/verbs/verbs_util.h | 14 |
1 files changed, 0 insertions, 14 deletions
diff --git a/tensorflow/contrib/verbs/verbs_util.h b/tensorflow/contrib/verbs/verbs_util.h index d9da396228..8b44adaedc 100644 --- a/tensorflow/contrib/verbs/verbs_util.h +++ b/tensorflow/contrib/verbs/verbs_util.h @@ -28,20 +28,6 @@ class TensorProto; class VerbsUtil { public: - // synchronous wrapper of CopyGPUTensorToCPU - static Status CopyGPUTensorToCPUSync(Device* gpu_device, - const DeviceContext* device_context, - const Tensor* gpu_tensor, - Tensor* cpu_tensor); - // synchronous wrapper of CopyCPUTensorToGPU - static Status CopyCPUTensorToGPUSync(const Tensor* cpu_tensor, - const DeviceContext* device_context, - Device* gpu_device, - Tensor* gpu_tensor); - // synchronous wrapper of SetProtoFromGPU - static Status SetProtoFromGPUSync(const Tensor& tensor, Device* dev, - const DeviceContext* device_context, - TensorProto* proto, bool is_dead); static string AppendStepidToKey(const string& key, int64 step_id); static void GetKeyAndStepId(const string& key_with_step_id, string& key, int64& step_id); |