diff --git a/test/gtest/bn_test_data.hpp b/test/gtest/bn_test_data.hpp index d3b1c6b073..276e59eb02 100644 --- a/test/gtest/bn_test_data.hpp +++ b/test/gtest/bn_test_data.hpp @@ -93,7 +93,10 @@ inline std::vector NetworkLarge() {64, 64, 112, 112, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0}, {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1}, {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1}, - {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0}}; + {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0}, + {128, 256, 14, 14, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1}, + {128, 256, 16, 16, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1}, + {670, 1, 224, 224, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1}}; } template <>