| 119 | } |
| 120 | |
| 121 | std::shared_ptr<Tensor> Network::get_io_tensor( |
| 122 | std::string name, LiteTensorPhase phase) { |
| 123 | LITE_ERROR_HANDLER_BEGIN |
| 124 | LITE_ASSERT(m_loaded, "get_io_tensor should be used after model loaded."); |
| 125 | LITE_CHECK_NON_NULL_POINTER(m_impl); |
| 126 | return m_impl->get_io_tensor(name, phase); |
| 127 | LITE_ERROR_HANDLER_END |
| 128 | } |
| 129 | |
| 130 | std::vector<std::shared_ptr<Tensor>> Network::get_discrete_tensors( |
| 131 | std::string name, LiteTensorPhase phase) { |
no outgoing calls