MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / add_extra_compile_output_spec

Method add_extra_compile_output_spec

src/opr/impl/loop/grad.cpp:290–298  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

288}
289
290void 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/*!

Callers 1

Calls 3

accum_state_sharedMethod · 0.80
makeFunction · 0.70
push_backMethod · 0.45

Tested by

no test coverage detected