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

Function get_fix_mask

dnn/test/common/dct_ref.cpp:10–17  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

8using Param = DctChannelSelectForward::Param;
9
10static inline FixCase get_fix_mask(Param::FastImpl impl) {
11 std::vector<int> fix_32_mask_offset{0, 16, 24, 32};
12 std::vector<int> fix_32_mask_val{0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32,
13 25, 18, 11, 4, 5, 0, 1, 8, 16, 9, 2,
14 3, 10, 0, 1, 8, 16, 9, 2, 3, 10};
15 megdnn_assert(impl == Param::FastImpl::FIX_32_MASK, "only support gen FIX_32_MASK");
16 return {fix_32_mask_offset, fix_32_mask_val};
17}
18
19CheckerHelper::TensorsConstriant gen_dct_constriant(
20 const size_t /* n */, const size_t ic, const size_t ih, const size_t iw,

Callers 2

gen_dct_constriantFunction · 0.85
gen_dct_caseFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected