Home
last modified time | relevance | path

Searched refs:ViaDMA (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.h43 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
Dcollective_rma_local.cc118 CopyTensor::ViaDMA("", // edge name (non-existent) in MemCpyAsync()
Drendezvous_mgr.cc110 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context, in SameWorkerRecvDone()
Dcopy_tensor.cc249 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA() function in tensorflow::CopyTensor
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc135 CopyTensor::ViaDMA("", // edge name (non-existent) in RecvFromPeer()
Dbase_rendezvous_mgr.cc281 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context, in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc302 tensorflow::CopyTensor::ViaDMA("copy", src_device_context, dst_device_context, in CopyToDevice()