| 153 | } |
| 154 | |
| 155 | std::shared_ptr<Tensor> Network::get_output_tensor(size_t index) { |
| 156 | LITE_ERROR_HANDLER_BEGIN |
| 157 | LITE_ASSERT(m_loaded, "get_output_tensor should be used after model loaded."); |
| 158 | LITE_CHECK_NON_NULL_POINTER(m_impl); |
| 159 | return m_impl->get_output_tensor(index); |
| 160 | LITE_ERROR_HANDLER_END |
| 161 | } |
| 162 | |
| 163 | Network& Network::set_async_callback(const AsyncCallback& callback) { |
| 164 | LITE_ERROR_HANDLER_BEGIN |
no outgoing calls