shl_ref_i805.h 3.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  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_REF_H_
  20. #define INCLUDE_SHL_I805_REF_H_
  21. #include "csi_nn.h"
  22. #include "shl_ref.h"
  23. int shl_i805_ref_conv2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  24. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  25. struct csinn_conv2d_params *params);
  26. int shl_i805_ref_conv2d_init_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  27. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  28. struct csinn_conv2d_params *params);
  29. int shl_i805_ref_depthwise_conv2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  30. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  31. struct csinn_conv2d_params *params);
  32. int shl_i805_ref_avgpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  33. struct csinn_pool_params *params);
  34. int shl_i805_ref_maxpool2d_init_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  35. struct csinn_pool_params *params);
  36. int shl_i805_ref_fullyconnected_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  37. struct csinn_tensor *weights, struct csinn_tensor *bias,
  38. struct csinn_fc_params *params);
  39. int shl_i805_ref_fullyconnected_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  40. struct csinn_tensor *weights, struct csinn_tensor *bias,
  41. struct csinn_fc_params *params);
  42. int shl_i805_ref_softmax_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  43. struct csinn_softmax_params *params);
  44. int shl_i805_ref_softmax_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  45. struct csinn_softmax_params *params);
  46. int shl_i805_ref_relu_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  47. struct csinn_relu_params *params);
  48. int shl_i805_ref_relu_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  49. struct csinn_relu_params *params);
  50. int shl_i805_ref_sigmoid_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  51. struct csinn_sigmoid_params *params);
  52. int shl_i805_ref_sigmoid_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  53. struct csinn_sigmoid_params *params);
  54. int shl_i805_ref_tanh_q7(struct csinn_tensor *input, struct csinn_tensor *output,
  55. struct csinn_siso_params *params);
  56. int shl_i805_ref_tanh_q15(struct csinn_tensor *input, struct csinn_tensor *output,
  57. struct csinn_siso_params *params);
  58. #endif // INCLUDE_SHL_I805_REF_H_