mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
@@ -20,7 +20,7 @@
|
||||
#include "gtest_utils.h"
|
||||
|
||||
namespace fastdeploy {
|
||||
#ifdef ENABLE_FDTENSOR_FUNC
|
||||
|
||||
TEST(fastdeploy, softmax) {
|
||||
FDTensor input, output;
|
||||
CheckShape check_shape;
|
||||
@@ -44,5 +44,5 @@ TEST(fastdeploy, softmax) {
|
||||
check_data(reinterpret_cast<const float*>(output.Data()),
|
||||
expected_result_axis1.data(), expected_result_axis1.size());
|
||||
}
|
||||
#endif
|
||||
|
||||
} // namespace fastdeploy
|
||||
Reference in New Issue
Block a user