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

Method MultipleInputOutput

src/gopt/test/subgraph_extractor.cpp:34–46  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

32MGB_DYN_TYPE_OBJ_FINAL_IMPL(MultipleInputOutput);
33
34MultipleInputOutput::MultipleInputOutput(
35 const VarNodeArray& inputs, const OperatorNodeConfig& config)
36 : Super(inputs[0]->owner_graph(), config, "multiple_input_output", inputs) {
37 for (auto&& i : inputs)
38 add_input({i});
39 if (inputs.size() == 1) {
40 add_output(None);
41 } else {
42 for (size_t i = 0; i < inputs.size(); ++i)
43 add_output(ssprintf("o%zu", i));
44 }
45 cg::add_workspace_output(this);
46}
47
48SymbolVarArray MultipleInputOutput::make(
49 const SymbolVarArray& inputs, const OperatorNodeConfig& config) {

Callers

nothing calls this directly

Calls 2

owner_graphMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected