functions.h 1.5 KB

123456789101112131415161718192021222324252627282930313233343536
  1. // Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #pragma once
  15. #include "ultra_infer/function/cast.h"
  16. #include "ultra_infer/function/clip.h"
  17. #include "ultra_infer/function/concat.h"
  18. #include "ultra_infer/function/cumprod.h"
  19. #include "ultra_infer/function/elementwise.h"
  20. #include "ultra_infer/function/full.h"
  21. #include "ultra_infer/function/gather_scatter_along_axis.h"
  22. #include "ultra_infer/function/gaussian_random.h"
  23. #include "ultra_infer/function/isfinite.h"
  24. #include "ultra_infer/function/linspace.h"
  25. #include "ultra_infer/function/math.h"
  26. #include "ultra_infer/function/pad.h"
  27. #include "ultra_infer/function/quantile.h"
  28. #include "ultra_infer/function/reduce.h"
  29. #include "ultra_infer/function/slice.h"
  30. #include "ultra_infer/function/softmax.h"
  31. #include "ultra_infer/function/sort.h"
  32. #include "ultra_infer/function/split.h"
  33. #include "ultra_infer/function/tile.h"
  34. #include "ultra_infer/function/transpose.h"