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

Method init_output_static_infer_desc

src/gopt/test/reformat_manager.cpp:325–329  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

323}
324
325void ReformatProfiler::MarkInputContiguous::init_output_static_infer_desc() {
326 using namespace cg::static_infer;
327 auto&& mgr = owner_graph()->static_infer_manager();
328 mgr.register_shape_infer(output(0), ShapeInferDesc::make_identity(input(0)));
329}
330
331void ReformatProfiler::MarkInputContiguous::add_input_layout_constraint() {
332 input(0)->add_layout_constraint_contiguous();

Callers

nothing calls this directly

Calls 1

register_shape_inferMethod · 0.45

Tested by

no test coverage detected