apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-11 11:30:20 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

[Other] Add namespace for functions (#538)

Browse Source 
Add namespace for functions
This commit is contained in:
Jason
2022-11-09 13:57:53 +08:00
committed by GitHub
parent 4706a7c32a
commit f2fed7959b
27 changed files with 63 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/text/ernie-3.0/cpp/seq_cls_infer.cc
View File

@@ -187,11 +187,11 @@ struct ErnieForSequenceClassificationPredictor {
std::vector<SeqClsResult>* seq_cls_results) {
const auto& logits = outputs[0];
fastdeploy::FDTensor probs;
fastdeploy::Softmax(logits, &probs);
fastdeploy::function::Softmax(logits, &probs);
fastdeploy::FDTensor labels, confidences;
fastdeploy::Max(probs, &confidences, {-1});
fastdeploy::ArgMax(probs, &labels, -1);
fastdeploy::function::Max(probs, &confidences, {-1});
fastdeploy::function::ArgMax(probs, &labels, -1);
if (labels.Numel() != confidences.Numel()) {
return false;
}