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

Merge pull request #45493 from advaitjain:tflm-flatbuffer-bazel

Browse Source 
PiperOrigin-RevId: 346447510
Change-Id: Ie4e7f7fa568d255f7aa6d2758577d4a935f7e7b9
This commit is contained in:
TensorFlower Gardener 2020-12-08 18:00:30 -08:00
commit a960cb2c42
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/lite/micro/tools/make/flatbuffers_download.sh
View File

@ -69,6 +69,15 @@ function patch_to_avoid_strtod() {
mv ${temp_flexbuffers_path} ${input_flexbuffers_path}
}
# The BUILD files in the downloaded folder result in an error with:
# bazel build tensorflow/lite/micro/...
#
# Parameters:
# $1 - path to the downloaded flatbuffers code.
function delete_build_files() {
rm -f `find ${1} -name BUILD`
}
DOWNLOADED_FLATBUFFERS_PATH=${DOWNLOADS_DIR}/flatbuffers
if [ -d ${DOWNLOADED_FLATBUFFERS_PATH} ]; then
@ -91,6 +100,8 @@ else
mv /tmp/flatbuffers-${ZIP_PREFIX} ${DOWNLOADED_FLATBUFFERS_PATH}
patch_to_avoid_strtod ${DOWNLOADED_FLATBUFFERS_PATH}/include/flatbuffers/flexbuffers.h
delete_build_files ${DOWNLOADED_FLATBUFFERS_PATH}
fi
echo "SUCCESS"