Skip to content

Commit

Permalink
[Functions] Fix functions comments (#1126)
Browse files Browse the repository at this point in the history
* fix functions comments

* Add paddle lite backend
  • Loading branch information
joey12300 authored Jan 12, 2023
1 parent c866377 commit f861095
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 0 deletions.
1 change: 1 addition & 0 deletions examples/text/ernie-3.0/cpp/seq_cls_infer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ bool CreateRuntimeOption(fastdeploy::RuntimeOption* option) {

if (FLAGS_device == "kunlunxin") {
option->UseKunlunXin();
option->UsePaddleLiteBackend();
return true;
} else if (FLAGS_device == "gpu") {
option->UseGpu();
Expand Down
4 changes: 4 additions & 0 deletions fastdeploy/function/transpose.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,10 @@
#include "fastdeploy/core/fd_tensor.h"

namespace fastdeploy {

/** \brief All C++ FDTensor Operation APIs are defined inside this namespace
*
*/
namespace function {
/** Excute the transpose operation for input FDTensor along given dims.
@param x The input tensor.
Expand Down

0 comments on commit f861095

Please sign in to comment.