Home
last modified time | relevance | path

Searched refs:CanUseDMA (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddma_helper.h26 static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); } in CanUseDMA() function
Dcopy_tensor.cc74 if (!DMAHelper::CanUseDMA(&from)) { in CopyHostToDevice()
139 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToHost()
203 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToDevice()
369 } else if (DMAHelper::CanUseDMA(&from)) { in WrappedTensorDeviceCopy()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc93 DMAHelper::CanUseDMA(&val) && dma_ok) { in GrpcRecvTensorAsync()
168 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc34 if (DMAHelper::CanUseDMA(&t) || t.dtype() == DT_VARIANT) { in OptionalDeviceCopy()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc248 if (!DMAHelper::CanUseDMA(&in) && in.dtype() != DT_VARIANT) { in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc101 if (!DMAHelper::CanUseDMA(&src)) { in PrepareCopy()
Dgpu_device.cc646 if (!DMAHelper::CanUseDMA(&from)) { in MaybeCopyTensorToGPU()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc573 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
/external/tensorflow/tensorflow/core/framework/
Dtensor.h644 bool CanUseDMA() const;
Dtensor.cc928 bool Tensor::CanUseDMA() const { in CanUseDMA() function in tensorflow::Tensor