aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_writer.py
diff options
context:
space:
mode:
authorFredrik Svedberg <fredrik.svedberg@arm.com>2020-09-04 09:46:17 +0200
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2020-09-07 06:23:54 +0000
commit835d8e10f33f411664cebe65d3f6a872f6cc849a (patch)
tree60c0cfd477cac7b604ade275ec82f54cbb14e9e8 /ethosu/vela/tflite_writer.py
parente5cf95b8c3de4e1e4cbc7046cafd4d84c7492596 (diff)
downloadethos-u-vela-835d8e10f33f411664cebe65d3f6a872f6cc849a.tar.gz
[MLBEDSW-2928] Add batching to softmax
Added batching to softmax by reshaping the input. Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com> Change-Id: I0b516f9bf2410fb86372b229beba4a7280c498cc
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
0 files changed, 0 insertions, 0 deletions