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

Method OutputCallback

imperative/src/impl/opr_utility.cpp:129–146  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

127MGB_DYN_TYPE_OBJ_FINAL_IMPL(OutputCallback);
128
129OutputCallback::OutputCallback(
130 Param param, const VarNodeArray& inputs, const OperatorNodeConfig& config)
131 : Super(inputs[0]->owner_graph(),
132 setup_config_cn(config, inputs[0]->comp_node()), "output_callback",
133 inputs),
134 m_param(std::move(param)) {
135 for (VarNode* i : inputs) {
136 add_input({i});
137 }
138 if (!m_param.borrow) {
139 input(0)->add_flag(VarNode::Flag::NO_SYS_STATIC_MEM_ALLOC);
140 }
141 add_output(None)
142 ->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE)
143 .add_flag(VarNode::Flag::NO_SYS_MEM_ALLOC)
144 .dtype(DType::from_enum(DTypeEnum::Byte));
145 add_equivalence_component<ScalarHash<void*>>(this);
146}
147
148SymbolVar OutputCallback::make(Param param, const SymbolVarArray& inputs) {
149 mgb_assert(inputs.size() >= 1);

Callers

nothing calls this directly

Calls 4

setup_config_cnFunction · 0.70
owner_graphMethod · 0.45
comp_nodeMethod · 0.45
dtypeMethod · 0.45

Tested by

no test coverage detected