Você não pode selecionar mais de 25 tópicos Os tópicos devem começar com uma letra ou um número, podem incluir traços ('-') e podem ter até 35 caracteres.
 
 
 
 
 

76 linhas
2.0 KiB

  1. /*
  2. * Copyright (C) 2010-2021 Arm Limited or its affiliates.
  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. /* ----------------------------------------------------------------------
  19. * Project: CMSIS NN Library
  20. * Title: arm_concatenation_s8_x.c
  21. * Description: s8 version of concatenation along the X axis
  22. *
  23. * $Date: October 2019
  24. * $Revision: V.1.0.0
  25. *
  26. * Target Processor: Cortex-M cores
  27. *
  28. * -------------------------------------------------------------------- */
  29. #include "arm_nnfunctions.h"
  30. #include "arm_nnsupportfunctions.h"
  31. /**
  32. * @ingroup groupNN
  33. */
  34. /**
  35. * @addtogroup Concatenation
  36. * @{
  37. */
  38. /*
  39. * s8 version of concatenation along the X axis
  40. *
  41. * Refer to header file for details.
  42. *
  43. */
  44. void arm_concatenation_s8_x(const int8_t *input,
  45. const uint16_t input_x,
  46. const uint16_t input_y,
  47. const uint16_t input_z,
  48. const uint16_t input_w,
  49. int8_t *output,
  50. const uint16_t output_x,
  51. const uint32_t offset_x)
  52. {
  53. const uint32_t num_iterations = input_y * input_z * input_w;
  54. output += offset_x;
  55. uint32_t i;
  56. // Copy per row
  57. for (i = 0; i < num_iterations; ++i)
  58. {
  59. arm_memcpy_q7(output, input, input_x);
  60. input += input_x;
  61. output += output_x;
  62. }
  63. }
  64. /**
  65. * @} end of Concatenation group
  66. */