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

Function output_canonizer

dnn/test/arm_common/lstm.cpp:14–18  ·  view source on GitHub ↗

in arm_common the reserve tensor is not used

Source from the content-addressed store, hash-verified

12namespace {
13//! in arm_common the reserve tensor is not used
14void output_canonizer(const CheckerHelper::TensorValueArray& arr) {
15 const TensorND& reserve = arr.back();
16 TensorND& modif_reserve = const_cast<TensorND&>(reserve);
17 modif_reserve.layout = TensorLayout();
18}
19} // namespace
20
21TEST_F(ARM_COMMON, LSTMCell) {

Callers

nothing calls this directly

Calls 2

TensorLayoutClass · 0.85
backMethod · 0.80

Tested by

no test coverage detected