GTEST_PARAMETER_TEST(ConcatGPU, LayerSemanticsGpu,
::testing::Values(semantic_concat_gpu));
-auto concat_dim3 = LayerGoldenTestParamType(
+auto concat_dim3_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=3"},
"2:3:3:2,2:3:3:3", "concat_dim3.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp32", "fp32");
-auto concat_dim2 = LayerGoldenTestParamType(
+auto concat_dim2_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=2"},
"2:3:2:3,2:3:3:3", "concat_dim2.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp32", "fp32");
-auto concat_dim1 = LayerGoldenTestParamType(
+auto concat_dim1_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=1"},
"2:2:3:3,2:3:3:3", "concat_dim1.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp32", "fp32");
GTEST_PARAMETER_TEST(ConcatGPU, LayerGoldenTest,
- ::testing::Values(concat_dim3, concat_dim2, concat_dim1));
+ ::testing::Values(concat_dim3_gpu, concat_dim2_gpu,
+ concat_dim1_gpu));
#ifdef ENABLE_FP16
-auto concat_dim3_w16a16 = LayerGoldenTestParamType(
+auto concat_dim3_w16a16_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=3"},
"2:3:3:2,2:3:3:3", "concat_dim3_w16a16.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp16", "fp16");
-auto concat_dim2_w16a16 = LayerGoldenTestParamType(
+auto concat_dim2_w16a16_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=2"},
"2:3:2:3,2:3:3:3", "concat_dim2_w16a16.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp16", "fp16");
-auto concat_dim1_w16a16 = LayerGoldenTestParamType(
+auto concat_dim1_w16a16_gpu = LayerGoldenTestParamType(
nntrainer::createLayer<nntrainer::ConcatLayerCl>, {"axis=1"},
"2:2:3:3,2:3:3:3", "concat_dim1_w16a16.nnlayergolden",
LayerGoldenTestParamOptions::SKIP_CALC_DERIV, "nchw", "fp16", "fp16");
GTEST_PARAMETER_TEST(ConcatGPU16, LayerGoldenTest,
- ::testing::Values(concat_dim3_w16a16, concat_dim2_w16a16,
- concat_dim1_w16a16));
+ ::testing::Values(concat_dim3_w16a16_gpu,
+ concat_dim2_w16a16_gpu,
+ concat_dim1_w16a16_gpu));
#endif