| 19 | namespace oneflow { |
| 20 | |
| 21 | void ChainKernelObserver::WillForward(KernelContext* kernel_ctx, const Kernel* kernel) { |
| 22 | for (const auto& observer : kernel_observers_) { observer->WillForward(kernel_ctx, kernel); } |
| 23 | } |
| 24 | |
| 25 | void ChainKernelObserver::DidForward(KernelContext* kernel_ctx, const Kernel* kernel) { |
| 26 | for (const auto& observer : kernel_observers_) { observer->DidForward(kernel_ctx, kernel); } |