| 270 | } |
| 271 | |
| 272 | int LITE_get_io_tensor( |
| 273 | LiteNetwork network, const char* io_name, LiteTensorPhase phase, |
| 274 | LiteTensor* tensor) { |
| 275 | LITE_CAPI_BEGIN(); |
| 276 | LITE_ASSERT(network, "The network pass to LITE api is null"); |
| 277 | auto io_tensor = |
| 278 | static_cast<lite::Network*>(network)->get_io_tensor(io_name, phase); |
| 279 | *tensor = io_tensor.get(); |
| 280 | LITE_CAPI_END(); |
| 281 | } |
| 282 | |
| 283 | int LITE_get_discrete_tensor( |
| 284 | LiteNetwork network, const char* io_name, size_t n_idx, LiteTensorPhase phase, |