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

Method init_output_static_infer_desc

src/opr/impl/utility.cpp:589–594  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

587}
588
589void VirtualGrad::init_output_static_infer_desc() {
590 using namespace cg::static_infer;
591 auto&& mgr = owner_graph()->static_infer_manager();
592 auto ovar = output(0), ivar = input(1);
593 mgr.register_shape_infer(ovar, ShapeInferDesc::make_identity(ivar));
594}
595
596void VirtualGrad::on_output_comp_node_stream_changed() {}
597

Callers

nothing calls this directly

Calls 6

make_constFunction · 0.85
is_static_var_shapeFunction · 0.85
register_shape_inferMethod · 0.45
infer_shape_fallibleMethod · 0.45
eq_shapeMethod · 0.45
to_stringMethod · 0.45

Tested by

no test coverage detected