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

Function apply_on_var_node

imperative/src/impl/ops/convolution.cpp:21–26  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

19}
20
21auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
22 auto&& conv = static_cast<const Convolution&>(def);
23 OperatorNodeConfig config{conv.make_name()};
24 return opr::Convolution::make(
25 inputs[0], inputs[1], conv.param(), conv.policy(), config);
26}
27
28std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
29 const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs) {

Callers

nothing calls this directly

Calls 12

output_dtypeMethod · 0.80
makeFunction · 0.70
make_nameMethod · 0.45
paramMethod · 0.45
nameMethod · 0.45
sizeMethod · 0.45
validMethod · 0.45
nodeMethod · 0.45
dtypeMethod · 0.45
comp_nodeMethod · 0.45
deduce_dtypeMethod · 0.45
opMethod · 0.45

Tested by

no test coverage detected