Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 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: f1cc7054f8
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'f1cc7054f8'
${ noResults }
mindspore/mindspore/lite/nnacl/int8/add_int8.h

56 lines
1.5 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_ADD_INT8_H_

  18. #define MINDSPORE_LITE_NNACL_ADD_INT8_H_

  19. 

  20. #include "nnacl/op_base.h"

  21. 

  22. typedef struct AddQuantParameter {

  23.   int left_shift_;

  24.   int32_t min_;

  25.   int32_t max_;

  26. 

  27.   int32_t in0_zp_;

  28.   int32_t in1_zp_;

  29.   int32_t out_zp_;

  30. 

  31.   int32_t in0_left_shift_;

  32.   int32_t in0_right_shift_;

  33.   int32_t in0_multiplier_;

  34. 

  35.   int32_t in1_left_shift_;

  36.   int32_t in1_right_shift_;

  37.   int32_t in1_multiplier_;

  38. 

  39.   int32_t out_left_shift_;

  40.   int32_t out_right_shift_;

  41.   int32_t out_multiplier_;

  42. } AddQuantParameter;

  43. 

  44. #ifdef __cplusplus

  45. extern "C" {

  46. #endif

  47. 

  48. void AddInt8(const int8_t *input0, const int8_t *input1, int8_t *output, int size, AddQuantParameter *params);

  49. void AddOptInt8(const int8_t *ptr_in, const int8_t element_in, int8_t *output, int size, AddQuantParameter *params);

  50. 

  51. #ifdef __cplusplus

  52. }

  53. #endif

  54. 

  55. #endif  // MINDSPORE_LITE_NNACL_ADD_INT8_H_