MNN/schema/generate.sh

48 lines
906 B
Bash
Raw Permalink Normal View History

2019-04-17 10:49:11 +08:00
#!/bin/bash
pushd "$(dirname $0)" > /dev/null
if [[ "$1" == "-lazy" ]] && [[ -d current ]]; then
popd > /dev/null
echo "*** done ***"
exit
fi
2019-04-17 10:49:11 +08:00
# check is flatbuffer installed or not
FLATC=../3rd_party/flatbuffers/tmp/flatc
if [ ! -e $FLATC ]; then
echo "*** building flatc ***"
# make tmp dir
pushd ../3rd_party/flatbuffers > /dev/null
[ ! -d tmp ] && mkdir tmp
cd tmp && rm -rf *
# build
2019-08-09 17:45:10 +08:00
cmake .. && cmake --build . --target flatc -- -j4
2019-04-17 10:49:11 +08:00
# dir recover
popd > /dev/null
fi
# determine directory to use
DIR="default"
if [ -d "private" ]; then
DIR="private"
fi
# 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 ***"
- build: - unify schema building in core and converter; - add more build script for android; - add linux build script for python; - ops impl: - add floor mod support in binary; - use eltwise impl in add/max/sub/mul binary for optimization; - remove fake double support in cast; - fix 5d support for concat; - add adjX and adjY support for batch matmul; - optimize conv2d back prop filter; - add pad mode support for conv3d; - fix bug in conv2d & conv depthwise with very small feature map; - optimize binary without broacast; - add data types support for gather; - add gather ND support; - use uint8 data type in gather v2; - add transpose support for matmul; - add matrix band part; - add dim != 4 support for padding, reshape & tensor convert; - add pad type support for pool3d; - make ops based on TensorFlow Lite quantization optional; - add all & any support for reduction; - use type in parameter as output type in reduction; - add int support for unary; - add variable weight support for conv2d; - fix conv2d depthwise weights initialization; - fix type support for transpose; - fix grad outputs count for reduce grad and reshape grad; - fix priorbox & detection output; - fix metal softmax error; - python: - add runSessionWithCallBackInfo interface; - add max nodes limit (1400) for visualization tool; - fix save error in python3; - align default dim; - convert: - add extra design for optimization; - add more post converting optimizers; - add caffe v1 weights blob support; - add cast, unary, conv transpose support for onnx model; - optimize batchnorm, conv with variable weights, prelu, reshape, slice, upsample for onnx model; - add cos/sin/atan/tan support for unary for tensorflow model; - add any/all support for reduction for tensorflow model; - add elu, conv3d, pool3d support for tensorflow model; - optimize argmax, batchnorm, concat, batch to space, conv with variable weights, prelu, slice for tensorflow model; - others: - fix size computer lock; - fix thread pool deadlock; - add express & parameters in express; - rewrite blitter chooser without static map; - add tests for expr;
2019-10-29 13:37:26 +08:00
find ../$DIR/*.fbs | xargs ../$FLATC -c -b --gen-object-api --reflect-names
2019-04-17 10:49:11 +08:00
popd > /dev/null
# finish
popd > /dev/null
2019-04-17 10:49:11 +08:00
echo "*** done ***"