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

Method add_input

src/opr/impl/loop/forward.cpp:19–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

17/* ========= FwdDesc ========= */
18
19SymbolVar LoopImpl::FwdDesc::add_input(SymbolVar inp, bool has_assign) {
20 if (!has_assign) {
21 auto&& var = m_input_no_assign_dedup[inp.node()];
22 if (var)
23 return var;
24 var = do_add_input(inp, {false, false}).node();
25 return var;
26 }
27 auto var = do_add_input(inp, {true, true});
28 m_input_assigned[var.node()] = false;
29 return var;
30}
31
32size_t LoopImpl::FwdDesc::add_output(SymbolVar val, OutputMode mode) {
33 auto ret = DescImplBase::add_output(val, mode);

Callers 15

TEST_FFunction · 0.45
TESTFunction · 0.45
TESTFunction · 0.45
TESTFunction · 0.45
TEST_FFunction · 0.45
test_basic_fwd_with_gradFunction · 0.45
TESTFunction · 0.45
test_null_gradFunction · 0.45
test_two_var_coupledFunction · 0.45
sin_by_taylorMethod · 0.45
exp_by_taylorMethod · 0.45
TESTFunction · 0.45

Calls 1

nodeMethod · 0.45

Tested by 13

TEST_FFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
TEST_FFunction · 0.36
test_basic_fwd_with_gradFunction · 0.36
TESTFunction · 0.36
test_null_gradFunction · 0.36
test_two_var_coupledFunction · 0.36
sin_by_taylorMethod · 0.36
exp_by_taylorMethod · 0.36
TESTFunction · 0.36