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

Function convert_to_lite_io

lite/lite-c/src/network.cpp:147–164  ·  view source on GitHub ↗

convert C NetworkIO io to lite::NetworkIO

Source from the content-addressed store, hash-verified

145
146//! convert C NetworkIO io to lite::NetworkIO
147lite::NetworkIO convert_to_lite_io(const LiteNetworkIO c_network_io) {
148 lite::NetworkIO network_io;
149 for (size_t i = 0; i < c_network_io.input_size; i++) {
150 LiteIO* c_io = c_network_io.inputs + i;
151 LITE_ASSERT(c_io->name, "input name of io tensor must set.");
152 network_io.inputs.push_back(
153 {c_io->name, static_cast<bool>(c_io->is_host), c_io->io_type,
154 convert_to_layout(c_io->config_layout)});
155 }
156 for (size_t i = 0; i < c_network_io.output_size; i++) {
157 LiteIO* c_io = c_network_io.outputs + i;
158 LITE_ASSERT(c_io->name, "output name of io tensor must set.");
159 network_io.outputs.push_back(
160 {c_io->name, static_cast<bool>(c_io->is_host), c_io->io_type,
161 convert_to_layout(c_io->config_layout)});
162 }
163 return network_io;
164}
165
166struct InnerIO {
167 std::vector<std::string> names;

Callers 2

LITE_make_networkFunction · 0.85

Calls 2

convert_to_layoutFunction · 0.85
push_backMethod · 0.45

Tested by

no test coverage detected