MNN/source/backend/tensorrt/execution/schema/generate.sh

19 lines
383 B
Bash
Executable File

#!/bin/bash
# check is flatbuffer installed or not
FLATC=/usr/local/bin/flatc
# clean up
echo "*** cleaning up ***"
rm -f current/*.h
[ ! -d current ] && mkdir current
# flatc all fbs
pushd current > /dev/null
echo "*** generating fbs under $DIR ***"
find ../*.fbs | xargs /usr/local/bin/flatc -c -b --gen-object-api --reflect-names
popd > /dev/null
# finish
echo "*** done ***"