experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix the arduino build.

Browse Source
This commit is contained in:
Advait Jain 2020-10-05 14:09:45 -07:00
parent 27c3083ae0
commit a4589c59e0
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/micro/kernels/fully_connected.h
View File

@ -24,7 +24,12 @@ namespace tflite {
// (reference or optimized) must define this function. // (reference or optimized) must define this function.
TfLiteRegistration Register_FULLY_CONNECTED(); TfLiteRegistration Register_FULLY_CONNECTED();
#if defined(CMSIS_NN) #if defined(CMSIS_NN) || defined(ARDUINO)
// The Arduino is a special case where we use the CMSIS kernels, but because of
// the current approach to building for Arduino, we do not support -DCMSIS_NN as
// part of the Arduino build. As a result, we use defined(ARDUINO) as a way to
// proxy for defined(CMSIS_NN) for this one special case.
// Returns a TfLiteRegistration struct for cmsis-nn kernel variant that only // Returns a TfLiteRegistration struct for cmsis-nn kernel variant that only
// supports int8. // supports int8.
TfLiteRegistration Register_FULLY_CONNECTED_INT8(); TfLiteRegistration Register_FULLY_CONNECTED_INT8();