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

Function init_output_static_infer_desc

src/core/test/graph/misc.cpp:75–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

73 }
74
75 void init_output_static_infer_desc() override {
76 using namespace mgb::cg::static_infer;
77 owner_graph()->static_infer_manager().register_shape_infer(
78 output(0),
79 {SourceType::CONSTANT, {}, [](TensorShape& dest, const InpVal&) {
80 dest = {1};
81 return true;
82 }});
83 }
84};
85
86MGB_DEFINE_OPR_CLASS(

Callers

nothing calls this directly

Calls 3

make_constFunction · 0.85
register_shape_inferMethod · 0.45
shapeMethod · 0.45

Tested by

no test coverage detected