convert C NetworkIO io to lite::NetworkIO
| 145 | |
| 146 | //! convert C NetworkIO io to lite::NetworkIO |
| 147 | lite::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 | |
| 166 | struct InnerIO { |
| 167 | std::vector<std::string> names; |
no test coverage detected