TEST_P(Test_Torch_layers, run_convolution)
{
// Output reference values are in range [23.4018, 72.0181]
- double l1 = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.08 : default_l1;
- double lInf = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? 0.43 : default_lInf;
+ double l1 = default_l1, lInf = default_lInf;
+ if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD)
+ {
+ l1 = 0.08;
- lInf = 0.42;
++ lInf = 0.43;
+ }
+ else if (target == DNN_TARGET_CUDA_FP16)
+ {
+ l1 = 0.08;
+ lInf = 0.5;
+ }
runTorchNet("net_conv", "", false, true, true, l1, lInf);
}
TEST_P(Test_Torch_layers, run_depth_concat)
{
- runTorchNet("net_depth_concat", "", false, true, true, 0.0,
- target == DNN_TARGET_OPENCL_FP16 ? 0.032 : 0.0);
+ double lInf = 0.0;
+ if (target == DNN_TARGET_OPENCL_FP16)
+ {
- lInf = 0.021;
++ lInf = 0.032;
+ }
+ else if (target == DNN_TARGET_CUDA_FP16)
+ {
+ lInf = 0.03;
+ }
+ runTorchNet("net_depth_concat", "", false, true, true, 0.0, lInf);
}
TEST_P(Test_Torch_layers, run_deconv)