Home
last modified time | relevance | path

Searched defs:send_dev_context (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc122 DeviceContext* send_dev_context, in CopyDeviceToDevice()
200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()
335 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc278 auto send_dev_context = in TEST_F() local
Dgrpc_worker_service.cc529 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync() local