| 288 | } |
| 289 | |
| 290 | void LoopGrad::AssignorGradOpr::add_extra_compile_output_spec( |
| 291 | ComputingGraph::OutputSpec& spec) { |
| 292 | mgb_assert(m_assignee_grads_buf_init); |
| 293 | auto ovar = output(0); |
| 294 | for (auto i : m_assignee_grads) { |
| 295 | auto updator = DepTensorUpdator::make(m_state->accum_state_shared(), i, ovar); |
| 296 | spec.push_back({updator, {}}); |
| 297 | } |
| 298 | } |
| 299 | |
| 300 | /* ==================== GradProxy ==================== */ |
| 301 | /*! |
no test coverage detected