Home
last modified time | relevance | path

Searched refs:send_dev_context (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc118 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost() argument
129 [edge_name, src, send_dev_context, out_allocator, status_cb, in CopyDeviceToHost()
136 src, to, send_dev_context, wrapped_done_); in CopyDeviceToHost()
150 send_dev_context->CopyDeviceTensorToCPU(&from, edge_name, src, to, in CopyDeviceToHost()
175 send_dev_context->CopyDeviceTensorToCPU(input, edge_name, src, output, in CopyDeviceToHost()
182 DeviceContext* send_dev_context, in CopyDeviceToDevice() argument
199 recv_dev_context, send_dev_context, out_allocator, status_cb, in CopyDeviceToDevice()
214 copy_function(send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
240 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
249 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA() argument
[all …]
Dcopy_tensor.h32 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
43 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc117 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync() local
123 CHECK(send_dev_context) in GrpcRecvTensorAsync()
136 send_dev_context->CopyDeviceTensorToCPU( in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc189 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy() argument
195 Status s = PrepareCopy(src, send_dev_context, *input, output, &dev_info, in DeviceToDeviceCopy()
202 static_cast<const GPUDeviceContext*>(send_dev_context) in DeviceToDeviceCopy()
247 send_dev_context->MaintainLifetimeOnStream(input, in DeviceToDeviceCopy()
Dgpu_util.h94 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc191 auto send_dev_context = in TEST_F() local
193 CHECK_EQ(123, send_dev_context->stream_id()); in TEST_F()
Dgrpc_worker_service.cc495 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync() local
501 CHECK(send_dev_context) in GrpcRecvTensorAsync()
514 send_dev_context->CopyDeviceTensorToCPU( in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1063 DeviceContext* send_dev_context = send_args.device_context; in RecvHandler() local
1064 CHECK(send_dev_context) in RecvHandler()
1079 StreamGPUOp(src_dev_, send_dev_context, in RecvHandler()
1094 src_dev_, send_dev_context, &in, &copy, in RecvHandler()