Searched refs:CanUseDMA (Results 1 – 10 of 10) sorted by relevance
26 static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); } in CanUseDMA() function
76 if (!DMAHelper::CanUseDMA(&from)) { in CopyHostToDevice()151 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToDevice()313 if (DMAHelper::CanUseDMA(&from)) { in WrappedTensorDeviceCopy()355 if (!DMAHelper::CanUseDMA(&from)) { in CopyDeviceToHost()
37 if (DMAHelper::CanUseDMA(&t) || t.dtype() == DT_VARIANT) { in OptionalDeviceCopy()
99 if (!DMAHelper::CanUseDMA(&src)) { in PrepareCopy()
642 if (!DMAHelper::CanUseDMA(&from)) { in MaybeCopyTensorToGPU()
240 if (!DMAHelper::CanUseDMA(&in) && in.dtype() != DT_VARIANT && in SameWorkerRecvDone()
154 TF_RET_CHECK(DMAHelper::CanUseDMA(input)); in TpuDeviceToDeviceCopy()
646 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
694 bool CanUseDMA() const;
980 bool Tensor::CanUseDMA() const { in CanUseDMA() function in tensorflow::Tensor