esp_nn.h 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. // Copyright 2020-2021 Espressif Systems (Shanghai) PTE LTD
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #pragma once
  15. #if defined(CONFIG_NN_OPTIMIZED)
  16. #ifdef CONFIG_IDF_TARGET_ESP32S3
  17. #define ARCH_ESP32_S3 1
  18. #endif
  19. #ifdef CONFIG_IDF_TARGET_ESP32
  20. #define ARCH_ESP32 1
  21. #endif
  22. #endif
  23. #ifdef __cplusplus
  24. extern "C" {
  25. #endif
  26. /* reference kernels included by default */
  27. #include "esp_nn_ansi_headers.h"
  28. #if defined(CONFIG_NN_OPTIMIZED)
  29. #ifdef ARCH_ESP32_S3
  30. #include "esp_nn_esp32s3.h"
  31. #endif
  32. #ifdef ARCH_ESP32
  33. #include "esp_nn_esp32.h"
  34. #endif
  35. #else
  36. #include "esp_nn_ansi_c.h"
  37. #endif
  38. #ifdef __cplusplus
  39. }
  40. #endif