Update IR documentation
Fix import in loop implementations