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

Function make_call_rec

src/opr/test/cond.cpp:38–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

36}
37
38SymbolVar make_call_rec(SymbolVar x, int* cnt) {
39 auto cb = [cnt](DeviceTensorND&) { ++*cnt; };
40 opr::CallbackInjector::Param param{cb};
41 param.invoke_for_static_infer = false;
42 return opr::CallbackInjector::make(x, param);
43}
44
45SymbolVar merge_one_out(
46 const SymbolVarArray& inputs_orig, MergeMode mode, size_t nr_distractor = 0,

Callers 4

test_merge_oprFunction · 0.85
test_simple_gradFunction · 0.85
test_nestedFunction · 0.85
TESTFunction · 0.85

Calls 1

makeFunction · 0.70

Tested by

no test coverage detected