File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
gallery/how_to/work_with_microtvm Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change 370370# the conversion. By default, TFLite keeps the inputs and outputs of our model as floats, so we must
371371# explicitly tell it to avoid this behavior.
372372
373+
373374def representative_dataset ():
374375 for image_batch , label_batch in full_dataset .take (10 ):
375376 yield [image_batch ]
@@ -563,6 +564,8 @@ def representative_dataset():
563564
564565shutil .rmtree (f"{ FOLDER } /models/project/build" , ignore_errors = True )
565566# sphinx_gallery_start_ignore
567+ from unittest .mock import MagicMock
568+
566569arduino_project = MagicMock ()
567570# sphinx_gallery_end_ignore
568571arduino_project .build ()
Original file line number Diff line number Diff line change @@ -84,6 +84,7 @@ IGNORED_WARNINGS=(
8484 ' autotvm:Cannot find config for target=llvm -keys=cpu -link-params=0'
8585 ' autotvm:One or more operators have not been tuned. Please tune your model for better performance. Use DEBUG logging level to see more details.'
8686 ' autotvm:Cannot find config for target=cuda -keys=cuda,gpu'
87+ ' absl:For model inputs containing unsupported operations'
8788)
8889
8990JOINED_WARNINGS=$( join_by ' |' " ${IGNORED_WARNINGS[@]} " )
You can’t perform that action at this time.
0 commit comments