zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: bc738d4ec3
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'bc738d4ec3'
${ noResults }
mindspore2022/mindspore/lite/nnacl/int8/arithmetic_self_int8.h

58 lines
2.1 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2020 Huawei Technologies Co., Ltd

  3.  *

  4.  * Licensed under the Apache License, Version 2.0 (the "License");

  5.  * you may not use this file except in compliance with the License.

  6.  * You may obtain a copy of the License at

  7.  *

  8.  * http://www.apache.org/licenses/LICENSE-2.0

  9.  *

  10.  * Unless required by applicable law or agreed to in writing, software

  11.  * distributed under the License is distributed on an "AS IS" BASIS,

  12.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  13.  * See the License for the specific language governing permissions and

  14.  * limitations under the License.

  15.  */

  16. 

  17. #ifndef MINDSPORE_LITE_NNACL_INT8_ARITHMETIC_SELF_INT8_H_

  18. #define MINDSPORE_LITE_NNACL_INT8_ARITHMETIC_SELF_INT8_H_

  19. 

  20. #ifdef ENABLE_NEON

  21. #include <arm_neon.h>

  22. #endif

  23. #include "nnacl/op_base.h"

  24. #include "nnacl/errorcode.h"

  25. #include "nnacl/quantization/quantize.h"

  26. 

  27. #ifdef __cplusplus

  28. extern "C" {

  29. #endif

  30. 

  31. int Int8ElementRound(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  32. 

  33. int Int8ElementFloor(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  34. 

  35. int Int8ElementCeil(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  36. 

  37. int Int8ElementAbs(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  38. 

  39. int Int8ElementSin(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  40. 

  41. int Int8ElementCos(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  42. 

  43. int Int8ElementLog(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  44. 

  45. int Int8ElementSqrt(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  46. 

  47. int Int8ElementRsqrt(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  48. 

  49. int Int8ElementSquare(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  50. 

  51. int Int8ElementLogicalNot(int8_t *input, int8_t *output, int element_size, ArithSelfQuantArg para);

  52. 

  53. #ifdef __cplusplus

  54. }

  55. #endif

  56. 

  57. #endif  // MINDSPORE_LITE_NNACL_INT8_ARITHMETIC_SELF_INT8_H_