| 1284 | } |
| 1285 | |
| 1286 | void LoopImpl::MutableStateSaver::update_subgraph_outspec( |
| 1287 | ComputingGraph::OutputSpec& spec) { |
| 1288 | for (auto&& i : m_var2info) { |
| 1289 | if (i.second.need_value) { |
| 1290 | spec.push_back({i.second.value_updator, {}}); |
| 1291 | } else if (i.second.need_shape) { |
| 1292 | spec.push_back({i.second.shape_updator, {}}); |
| 1293 | } |
| 1294 | } |
| 1295 | } |
| 1296 | |
| 1297 | void LoopImpl::MutableStateSaver::on_fwd_begin() { |
| 1298 | if (!m_enabled) |