diff options
author | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-10-14 10:57:46 +0000 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-10-14 10:57:46 +0000 |
commit | 1068362a83dd15613561189c00e364cddfd3071d (patch) | |
tree | 05d09cdc84c2da6e8a5e9b33e15e3deda1a61219 /ethosu/vela/tflite_writer.py | |
parent | 04986c0016e59993563490fe67052371fc0e1ad2 (diff) | |
download | ethos-u-vela-1068362a83dd15613561189c00e364cddfd3071d.tar.gz |
Revert "MLBEDSW-3219: Suppress CPU info for Const/Placeholder"
This reverts commit 04986c0016e59993563490fe67052371fc0e1ad2.
Reason for revert: Merged by mistake
Change-Id: I150ad9ba7074ad1e80f21180aeba56a454d9f748
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
-rw-r--r-- | ethosu/vela/tflite_writer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index 4cb5df17..e24aa282 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -77,7 +77,7 @@ class TFLiteSerialiser: self.scratch_fast_buf_id = 1 # Always assign scratch_fast to buffer 1 self.buffers_to_write = [] # have an empty array there - self.ops_to_ignore = Op.op_set(Op.is_startup_init_op) + self.ops_to_ignore = set((Op.Const, Op.Placeholder, Op.SubgraphInput)) self.tensors_to_reshape = {} |