Searched refs:send_dev_context (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 122 DeviceContext* send_dev_context, in CopyDeviceToDevice() argument
139 recv_dev_context, send_dev_context, out_allocator, status_cb, in CopyDeviceToDevice()
146 send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
162 copy_function(send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
191 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA() argument
235 send_dev_context, recv_dev_context, src, dst, in ViaDMA()
268 cpu_tensor, send_dev_context, in ViaDMA()
277 output, send_dev_context, std::move(done)); in ViaDMA()
335 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost() argument
[all …]
|
| D | copy_tensor.h | 32 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
43 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
75 DeviceContext* send_dev_context, StatusCallback done);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy() argument
194 Status s = PrepareCopy(src, send_dev_context, *input, output, &dev_info, in DeviceToDeviceCopy()
201 static_cast<const GPUDeviceContext*>(send_dev_context) in DeviceToDeviceCopy()
246 send_dev_context->MaintainLifetimeOnStream(input, in DeviceToDeviceCopy()
|
| D | gpu_util.h | 94 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 278 auto send_dev_context = in TEST_F() local
280 CHECK_EQ(123, send_dev_context->stream_id()); in TEST_F()
|
| D | grpc_worker_service.cc | 529 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync() local
535 CHECK(send_dev_context) in GrpcRecvTensorAsync()
548 src_dev, copy, send_dev_context, copy_ready); in GrpcRecvTensorAsync()
|