Method register_value_infer

src/core/impl/graph/static_infer_impl.cpp:824–835

}

void StaticInferManagerImpl::register_value_infer(
        Tag dest, const ValueInferDesc& 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 not inserted before value desc");
    mgb_assert(!t.value, "value infer already registered");
    t.value = m_mem_pool_value_trait.alloc(dest);
    t.value->init(desc, this);
}

InferType StaticInferManagerImpl::get_infer_type(Tag dest) {
    return get_tag_trait_container(dest).get_infer_type();