Searched refs:NeedWait (Results 1 – 9 of 9) sorted by relevance
67 if (input->NeedWait()) { in IsTaskReady()83 if (tensor->NeedWait()) { in WaitLockedInputs()94 if (input_tensor->NeedWait() && !input_tensor->IsGraphOutput()) { in WaitLockedInputs()435 if (tensor->NeedWait()) { in RunOp()456 if (tensor->NeedWait()) { in RunOp()
26 virtual bool NeedWait() = 0;
334 bool NeedWait() const { return need_wait_; } in NeedWait() function354 return device_event_->NeedWait(); in NeedWaitDevice()
31 bool NeedWait() override;
59 bool GpuEvent::NeedWait() { return need_wait_; } in NeedWait() function in mindspore::device::gpu::GpuEvent
90 bool AscendEvent::NeedWait() { return need_wait_; } in NeedWait() function in mindspore::device::ascend::AscendEvent
329 if (tensor.NeedWait()) { in FlushFromCache()367 if (tensor.NeedWait()) { in SyncAsNumpy()
46 if (tensor->NeedWait()) { in TensorToPyData()