Searched refs:CanUseDMA (Results 1 – 8 of 8) sorted by relevance
26 static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); } in CanUseDMA() function
68 if (!DMAHelper::CanUseDMA(&from)) { in CopyHostToDevice()126 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToHost()189 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToDevice()
96 DMAHelper::CanUseDMA(&val) && dma_ok) { in GrpcRecvTensorAsync()
103 if (!DMAHelper::CanUseDMA(&src)) { in PrepareCopy()
546 if (!DMAHelper::CanUseDMA(&from)) { in MaybeCopyTensorToGPU()
248 if (!DMAHelper::CanUseDMA(&in) && in.dtype() != DT_VARIANT) { in SameWorkerRecvDone()
501 bool CanUseDMA() const;
878 bool Tensor::CanUseDMA() const { in CanUseDMA() function in tensorflow::Tensor