shl_i805.h 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. /*
  2. * Copyright (C) 2016-2022 T-Head Semiconductor Co., Ltd. All rights reserved.
  3. *
  4. * SPDX-License-Identifier: Apache-2.0
  5. *
  6. * Licensed under the Apache License, Version 2.0 (the License); you may
  7. * not use this file except in compliance with the License.
  8. * You may obtain a copy of the License at
  9. *
  10. * www.apache.org/licenses/LICENSE-2.0
  11. *
  12. * Unless required by applicable law or agreed to in writing, software
  13. * distributed under the License is distributed on an AS IS BASIS, WITHOUT
  14. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. * See the License for the specific language governing permissions and
  16. * limitations under the License.
  17. */
  18. /* CSI-NN2 version 2.0.x */
  19. #ifndef INCLUDE_SHL_I805_H_
  20. #define INCLUDE_SHL_I805_H_
  21. #include <stdbool.h>
  22. #include <stdint.h>
  23. #include <stdlib.h>
  24. #include <string.h>
  25. #include "csi_nn.h"
  26. #include "shl_ref.h"
  27. int shl_i805_conv2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  28. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  29. struct csinn_conv2d_params *params);
  30. int shl_i805_conv2d_init_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  31. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  32. struct csinn_conv2d_params *params);
  33. int shl_i805_depthwise_conv2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  34. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  35. struct csinn_conv2d_params *params);
  36. int shl_i805_avgpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  37. struct csinn_pool_params *params);
  38. int shl_i805_maxpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  39. struct csinn_pool_params *params);
  40. int shl_i805_fullyconnected_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  41. struct csinn_tensor *weights, struct csinn_tensor *bias,
  42. struct csinn_fc_params *params);
  43. int shl_i805_fullyconnected_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  44. struct csinn_tensor *weights, struct csinn_tensor *bias,
  45. struct csinn_fc_params *params);
  46. int shl_i805_softmax_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  47. struct csinn_softmax_params *params);
  48. int shl_i805_softmax_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  49. struct csinn_softmax_params *params);
  50. int shl_i805_relu_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  51. struct csinn_relu_params *params);
  52. int shl_i805_relu_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  53. struct csinn_relu_params *params);
  54. int shl_i805_sigmoid_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  55. struct csinn_sigmoid_params *params);
  56. int shl_i805_sigmoid_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  57. struct csinn_sigmoid_params *params);
  58. int shl_i805_tanh_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  59. struct csinn_siso_params *params);
  60. int shl_i805_tanh_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  61. struct csinn_siso_params *params);
  62. /*********************** u8 asym quant opt func *********************************/
  63. int shl_i805_add_init_u8(struct csinn_tensor *input0, struct csinn_tensor *input1,
  64. struct csinn_tensor *output, struct csinn_diso_params *params);
  65. int shl_i805_add_u8(struct csinn_tensor *input0, struct csinn_tensor *input1,
  66. struct csinn_tensor *output, struct csinn_diso_params *params);
  67. int shl_i805_clip_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  68. struct csinn_clip_params *params);
  69. int shl_i805_clip_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  70. struct csinn_clip_params *params);
  71. int shl_i805_conv2d_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  72. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  73. struct csinn_conv2d_params *params);
  74. int shl_i805_conv2d_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  75. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  76. struct csinn_conv2d_params *params);
  77. int shl_i805_depthwise_conv2d_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  78. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  79. struct csinn_conv2d_params *params);
  80. int shl_i805_depthwise_conv2d_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  81. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  82. struct csinn_conv2d_params *params);
  83. int shl_i805_fullyconnected_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  84. struct csinn_tensor *weights, struct csinn_tensor *bias,
  85. struct csinn_fc_params *params);
  86. int shl_i805_fullyconnected_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  87. struct csinn_tensor *weights, struct csinn_tensor *bias,
  88. struct csinn_fc_params *params);
  89. int shl_i805_maxpool2d_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  90. struct csinn_pool_params *params);
  91. int shl_i805_mul_init_u8(struct csinn_tensor *input0, struct csinn_tensor *input1,
  92. struct csinn_tensor *output, struct csinn_diso_params *params);
  93. int shl_i805_mul_u8(struct csinn_tensor *input0, struct csinn_tensor *input1,
  94. struct csinn_tensor *output, struct csinn_diso_params *params);
  95. int shl_i805_relu_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  96. struct csinn_relu_params *params);
  97. int shl_i805_relu_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  98. struct csinn_relu_params *params);
  99. int shl_i805_relu6_init_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  100. struct csinn_relu_params *params);
  101. int shl_i805_relu6_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  102. struct csinn_relu_params *params);
  103. int shl_i805_reshape_u8(struct csinn_tensor *input, struct csinn_tensor *output,
  104. struct csinn_reshape_params *params);
  105. #endif // INCLUDE_SHL_I805_H_