MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / init_output_static_infer_desc

Method init_output_static_infer_desc

src/jit/impl/executor_opr.cpp:137–148  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

135}
136
137void JITExecutor::init_output_static_infer_desc() {
138 using namespace cg::static_infer;
139 auto&& mgr = owner_graph()->static_infer_manager();
140 mgr.register_shape_infer(
141 output(0), ShapeInferDesc::make_identity(m_internal_graph->shape_infer()));
142 m_compiler->init_workspace_size_infer(this);
143 if (m_internal_graph->value_infer()) {
144 mgr.register_value_infer(
145 output(0),
146 ValueInferDesc::make_identity(m_internal_graph->value_infer()));
147 }
148}
149
150void JITExecutor::scn_do_execute() {
151 if (m_executable == nullptr || m_args.need_update) {

Callers

nothing calls this directly

Calls 5

shape_inferMethod · 0.80
value_inferMethod · 0.80
register_shape_inferMethod · 0.45
register_value_inferMethod · 0.45

Tested by

no test coverage detected