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

Method Reshape

src/opr/impl/tensor_manip.cpp:334–341  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

332MGB_DYN_TYPE_OBJ_FINAL_IMPL(Reshape);
333
334Reshape::Reshape(
335 VarNode* inp, VarNode* tshp, Param unspec_axis,
336 const OperatorNodeConfig& config)
337 : Super{inp->owner_graph(), config, "reshape", {inp}},
338 m_unspec_axis{unspec_axis} {
339 reshapebrdcast_init(inp, tshp);
340 add_equivalence_component<PODHash<Param>>(&m_unspec_axis);
341}
342
343SymbolVar Reshape::make(
344 SymbolVar inp, SymbolVar tshp, Param unspec_axis,

Callers 4

make_full_if_noneFunction · 0.80
syncbn_split_statsFunction · 0.80
_make_full_if_noneFunction · 0.80
einsumFunction · 0.80

Calls 1

owner_graphMethod · 0.45

Tested by

no test coverage detected