mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
@@ -21,6 +21,7 @@
|
||||
#include "fastdeploy/utils/utils.h"
|
||||
|
||||
namespace fastdeploy {
|
||||
namespace function {
|
||||
template <typename T>
|
||||
struct ValueClip {
|
||||
T operator()(const T& x) const {
|
||||
@@ -124,4 +125,5 @@ void Softmax(const FDTensor& x, FDTensor* out, int axis) {
|
||||
([&] { SoftmaxKernel<data_t>(x, &out_tmp, axis); }));
|
||||
*out = std::move(out_tmp);
|
||||
}
|
||||
} // namespace function
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user