Method map_var

src/opr/impl/loop/grad.cpp:413–413

get var in grad graph corresponding to given var in fwd graph

    //! get var in grad graph corresponding to given var in fwd graph
    VarNode* map_var(VarNode* fwd_var) const { return m_var_fwd2grad.at(fwd_var); }

    //! map from assignee var in subgraph to AssigneeInfo
    auto&& assignee2info() const { return m_assignee2info; }