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

Function apply_on_var_node

imperative/src/impl/ops/concatenate.cpp:81–86  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

79}
80
81auto apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
82 auto&& op = static_cast<const Concat&>(def);
83 cg::OperatorNodeConfig config{op.comp_node};
84 config.name(op.make_name());
85 return opr::Concat::make(inputs, op.axis, config);
86}
87
88std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
89 const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs) {

Callers

nothing calls this directly

Calls 5

emplace_backMethod · 0.80
makeFunction · 0.70
nameMethod · 0.45
make_nameMethod · 0.45
nodeMethod · 0.45

Tested by

no test coverage detected