| 136 | } |
| 137 | |
| 138 | void NetworkImplDft::set_memory_allocator(std::shared_ptr<Allocator> user_allocator) { |
| 139 | auto allocator = std::make_shared<UserStaticMemAlloc>(user_allocator); |
| 140 | LITE_ASSERT(m_load_config.comp_graph); |
| 141 | m_load_config.comp_graph->set_device_memory_allocator(allocator); |
| 142 | } |
| 143 | |
| 144 | //! share the runtime memory with other network, the weights is not shared |
| 145 | void NetworkImplDft::share_runtime_memory_with(Network::NetworkImplBase* network_impl) { |
nothing calls this directly
no test coverage detected