shl_asp.h 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  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_ASP_H_
  20. #define INCLUDE_SHL_ASP_H_
  21. #include "csi_nn.h"
  22. #include "shl_ref.h"
  23. int shl_asp_avgpool2d(struct csinn_tensor *input, struct csinn_tensor *output,
  24. struct csinn_pool_params *params);
  25. int shl_asp_conv2d(struct csinn_tensor *input, struct csinn_tensor *output,
  26. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  27. struct csinn_conv2d_params *params);
  28. int shl_asp_depthwise_conv2d(struct csinn_tensor *input, struct csinn_tensor *output,
  29. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  30. struct csinn_conv2d_params *params);
  31. int shl_asp_fullyconnected(struct csinn_tensor *input, struct csinn_tensor *output,
  32. struct csinn_tensor *kernel, struct csinn_tensor *bias,
  33. struct csinn_fc_params *params);
  34. int shl_asp_maxpool2d(struct csinn_tensor *input, struct csinn_tensor *output,
  35. struct csinn_pool_params *params);
  36. #endif // INCLUDE_SHL_ASP_H_