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

Method generate

src/jit/impl/internal_graph.cpp:157–164  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

155}
156
157InternalGraphPtr InternalGraphGenerator::generate() {
158 m_input_idx = 0;
159
160 auto new_nd = replace_graph_by_placeholder();
161 auto igraph = std::make_shared<InternalGraph>(
162 new_nd, m_output, m_output, to_placeholder_opr_arr(m_placeholders));
163 return expand_executor_opr(igraph);
164}
165
166size_t InternalGraphGenerator::get_cnt_input_if_add(cg::OperatorNodeBase* opr) const {
167 // minus 1 first because this opr should be removed from subgraph's input

Callers 14

apply_on_var_nodeFunction · 0.45
run<expand_jit_executor>Function · 0.45
TESTFunction · 0.45
ensure_init_graphMethod · 0.45
run<simple>Function · 0.45
run<grad>Function · 0.45
run_mlirFunction · 0.45
run_mlir_broadcastFunction · 0.45
run_mlir_different_shapeFunction · 0.45
run_mlir_modeFunction · 0.45
run_typecvtFunction · 0.45
run_dimshuffleFunction · 0.45

Calls 1

expand_executor_oprFunction · 0.85

Tested by 12

run<expand_jit_executor>Function · 0.36
TESTFunction · 0.36
ensure_init_graphMethod · 0.36
run<simple>Function · 0.36
run<grad>Function · 0.36
run_mlirFunction · 0.36
run_mlir_broadcastFunction · 0.36
run_mlir_different_shapeFunction · 0.36
run_mlir_modeFunction · 0.36
run_typecvtFunction · 0.36
run_dimshuffleFunction · 0.36
TESTFunction · 0.36