Function TEST_F

dnn/test/arm_common/rnn.cpp:11–39

using namespace test;

TEST_F(ARM_COMMON, RNNCell) {
    Checker<RNNCell> checker(handle());
    using NonlineMode = param::RNNCell::NonlineMode;
    param::RNNCell param;
    for (auto mode : {NonlineMode::IDENTITY, NonlineMode::RELU, NonlineMode::TANH})
        for (size_t batch : {1, 4})
            for (size_t n : {3, 4, 5, 23, 100})
                for (size_t h : {5, 23, 100})
                    for (size_t out : {3, 6, 25, 100}) {
                        param.nonlineMode = mode;
                        checker.set_param(param);
                        checker.exec(
                                {{batch, n},
                                 {out, n},
                                 {1, out},
                                 {batch, h},
                                 {out, h},
                                 {1, out},
                                 {}});
                        checker.exec(
                                {{batch, n},
                                 {out, n},
                                 {batch, out},
                                 {batch, h},
                                 {out, h},
                                 {batch, out},
                                 {}});
                    }
}

TEST_F(ARM_COMMON, RNNCellRecord) {
    TaskRecordChecker<RNNCell> checker(0);