[Other] Enable to hidden INFO/WARNING log informations (#1368)

* Enable to hidden INFO/WARNING log informations

* Fix build error

---------

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
This commit is contained in:
Jason
2023-02-20 17:24:12 +08:00
committed by GitHub
parent b6658b8094
commit dc5b4be7a9
5 changed files with 36 additions and 5 deletions

View File

@@ -26,7 +26,8 @@ namespace function {
@param out The output tensor which stores the result.
@param axis Axis which will be gathered.
*/
void GatherAlongAxis(const FDTensor& x, const FDTensor& index, FDTensor* result,
FASTDEPLOY_DECL void GatherAlongAxis(const FDTensor& x,
const FDTensor& index, FDTensor* result,
int axis);
} // namespace function