Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcopy_tensor.cc124 DeviceContext* send_dev_context, in CopyDeviceToDevice()
202 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()
343 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_util.cc111 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util.cc188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Drpc_rendezvous_mgr_test.cc282 auto send_dev_context = in TEST_F() local
H A Dgrpc_worker_service.cc534 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync() local