shl_e804.h 3.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  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_E804_H_
  20. #define INCLUDE_SHL_E804_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_e804_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_e804_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_e804_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_e804_avgpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  37. struct csinn_pool_params *params);
  38. int shl_e804_maxpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  39. struct csinn_pool_params *params);
  40. int shl_e804_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_e804_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_e804_softmax_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  47. struct csinn_softmax_params *params);
  48. int shl_e804_softmax_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  49. struct csinn_softmax_params *params);
  50. int shl_e804_relu_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  51. struct csinn_relu_params *params);
  52. int shl_e804_relu_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  53. struct csinn_relu_params *params);
  54. int shl_e804_sigmoid_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  55. struct csinn_sigmoid_params *params);
  56. int shl_e804_sigmoid_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  57. struct csinn_sigmoid_params *params);
  58. int shl_e804_tanh_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  59. struct csinn_siso_params *params);
  60. int shl_e804_tanh_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  61. struct csinn_siso_params *params);
  62. #endif // INCLUDE_SHL_E804_H_