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

Function apply_on_var_node

imperative/src/impl/ops/matrix_inverse.cpp:10–15  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

8
9namespace {
10auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
11 auto&& op = def.cast_final_safe<MatrixInverse>();
12 mgb_assert(inputs.size() == 1);
13 OperatorNodeConfig config{op.make_name()};
14 return opr::MatrixInverse::make(inputs[0], {}, config);
15}
16OP_TRAIT_REG(MatrixInverse, MatrixInverse)
17 .apply_on_var_node(apply_on_var_node)
18 .fallback();

Callers

nothing calls this directly

Calls 3

makeFunction · 0.70
sizeMethod · 0.45
make_nameMethod · 0.45

Tested by

no test coverage detected