Method register_shape_infer

src/core/impl/graph/static_infer_impl.cpp:801–822

}

void StaticInferManagerImpl::register_shape_infer(
        Tag dest, const ShapeInferDesc& desc) {
    mgb_assert(dest->owner_opr() == m_register_allowed_opr);
    for (auto&& i : desc.deps)
        mgb_assert(dest->owner_graph() == i.dest->owner_graph());

    auto&& t = get_tag_trait_container(dest);
    mgb_assert(!t.shape, "shape desc already inserted");
    auto ptr = m_mem_pool_shape_trait.alloc_unique(dest);
    ptr->init(desc, this);
    if (ptr->infer_type() == InferType::CONST &&
        !dest->contain_flag(VarNode::Flag::VOLATILE_CONTENT)) {
        // infer const shapes immediately
        auto r = ptr->infer(false, false);
        mgb_assert(r && r->m_shape);
        dest->shape(r->shape());
        dest->m_allow_shape_change = false;
        t.shape = m_mem_pool_const_shape_trait.alloc(dest);
    } else {
        t.shape = ptr.release();
    }
}

void StaticInferManagerImpl::register_value_infer(
        Tag dest, const ValueInferDesc& desc) {