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

Method set_io

lite/src/mge/network_impl.cpp:669–676  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

667}
668
669void NetworkImplDft::set_io(const NetworkIO& network_io) {
670 for (auto&& in : network_io.inputs) {
671 m_network_io->inputs.emplace_back(in);
672 }
673 for (auto&& out : network_io.outputs) {
674 m_network_io->outputs.emplace_back(out);
675 }
676}
677
678void NetworkImplDft::try_infer_tensor_layout(std::shared_ptr<Tensor> tensor, Var var) {
679 using InferType = mgb::cg::static_infer::InferType;

Callers 3

network.cppFile · 0.80
NetworkMethod · 0.80
prase_modelMethod · 0.80

Calls 1

emplace_backMethod · 0.80

Tested by

no test coverage detected