Function apply_on_var_node

imperative/src/impl/ops/rnn.cpp:9–15

namespace rnn_cell {
auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
    auto&& op = static_cast<const RNNCell&>(def);
    mgb_assert(inputs.size() == 6);
    return opr::RNNCell::make(
            inputs[0], inputs[1], inputs[2], inputs[3], inputs[4], inputs[5],
            op.param());
}
OP_TRAIT_REG(RNNCell, RNNCell).apply_on_var_node(apply_on_var_node).fallback();
}  // namespace rnn_cell