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

Function TEST_F

dnn/test/x86/conv_bias.cpp:16–33  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

14namespace test {
15
16TEST_F(X86, CONV_BIAS_FORWARD) {
17 using namespace conv_bias;
18 std::vector<TestArg> args = get_args();
19 Checker<ConvBiasForward> checker(handle());
20 NormalRNG default_rng;
21 ConstValue const_val;
22 for (auto&& arg : args) {
23 checker.set_dtype(0, dtype::Float32())
24 .set_dtype(1, dtype::Float32())
25 .set_dtype(2, dtype::Float32())
26 .set_rng(0, &default_rng)
27 .set_rng(1, &default_rng)
28 .set_rng(2, &default_rng)
29 .set_epsilon(1e-3)
30 .set_param(arg.param)
31 .execs({arg.src, arg.filter, arg.bias, {}, {}});
32 }
33}
34
35TEST_F(X86, CONV_BIAS_FORWARD_RECORD) {
36 using namespace conv_bias;

Callers

nothing calls this directly

Calls 15

cb2Function · 0.85
get_conv_bias_1x1_argsFunction · 0.85
check_conv_biasFunction · 0.85
benchmark_impl_compFunction · 0.85
set_dtypeMethod · 0.80

Tested by

no test coverage detected