Searched refs:CanUseDMA (Results 1 – 10 of 10) sorted by relevance
26 static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); } in CanUseDMA() function
74 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()
93 DMAHelper::CanUseDMA(&val) && dma_ok) { in GrpcRecvTensorAsync()168 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
34 if (DMAHelper::CanUseDMA(&t) || t.dtype() == DT_VARIANT) { in OptionalDeviceCopy()
248 if (!DMAHelper::CanUseDMA(&in) && in.dtype() != DT_VARIANT) { in SameWorkerRecvDone()
101 if (!DMAHelper::CanUseDMA(&src)) { in PrepareCopy()
646 if (!DMAHelper::CanUseDMA(&from)) { in MaybeCopyTensorToGPU()
573 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
644 bool CanUseDMA() const;
928 bool Tensor::CanUseDMA() const { in CanUseDMA() function in tensorflow::Tensor