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

Method wrap_outputs

imperative/src/impl/transformations/format.cpp:91–98  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

89}
90
91inline ValueRefList FormatTransformation::wrap_outputs(
92 const ValueRefList& outputs, Format format) const {
93 ValueRefList wrapped_outputs(outputs.size());
94 for (size_t i = 0; i < outputs.size(); ++i) {
95 wrapped_outputs[i] = wrap_output(outputs[i], format);
96 }
97 return wrapped_outputs;
98}
99
100inline bool FormatTransformation::check_all_format_value(
101 const Span<ValueRef>& inputs) const {

Callers 9

dimshuffle_ruleFunction · 0.80
reshape_ruleFunction · 0.80
broadcast_ruleFunction · 0.80
subtensor_ruleFunction · 0.80
indexing_ruleFunction · 0.80
setsubtensor_ruleFunction · 0.80
elemwise_ruleFunction · 0.80
concat_ruleFunction · 0.80
identity_rule_helperFunction · 0.80

Calls 1

sizeMethod · 0.45

Tested by

no test coverage detected