diff options
Diffstat (limited to 'tests/datasets_new/system_tests/alexnet')
-rw-r--r-- | tests/datasets_new/system_tests/alexnet/AlexNetActivationLayerDataset.h | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tests/datasets_new/system_tests/alexnet/AlexNetActivationLayerDataset.h b/tests/datasets_new/system_tests/alexnet/AlexNetActivationLayerDataset.h index 27e9956647..7062c2e6b8 100644 --- a/tests/datasets_new/system_tests/alexnet/AlexNetActivationLayerDataset.h +++ b/tests/datasets_new/system_tests/alexnet/AlexNetActivationLayerDataset.h @@ -44,10 +44,7 @@ public: AlexNetActivationLayerDataset() : CartesianProductDataset { - framework::dataset::make("Shape", { - TensorShape(55U, 55U, 96U), TensorShape(27U, 27U, 256U), - TensorShape(13U, 13U, 384U), TensorShape(13U, 13U, 256U), - TensorShape(4096U) }), + framework::dataset::make("Shape", { TensorShape(55U, 55U, 96U), TensorShape(27U, 27U, 256U), TensorShape(13U, 13U, 384U), TensorShape(13U, 13U, 256U), TensorShape(4096U) }), framework::dataset::make("Info", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)) } { |