diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-11-27 12:23:42 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-11-27 14:57:24 +0000 |
| commit | 14e8a2041877f277f96fe339f48d4f51e0395a42 (patch) | |
| tree | 778b399efa01874c8cebece32ff5c9e7b9602739 /ethosu/vela/vela.py | |
| parent | 061eeb4c6a1af7780992ae6d3967905b03e2433a (diff) | |
| download | ethos-u-vela-14e8a2041877f277f96fe339f48d4f51e0395a42.tar.gz | |
vela: Rename --keep-scale-placement CLI
- Changed to --cache-bias-scale-tensor
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I285fe253f03ba98eff36dbe996ad3a57e2ee3d99
Diffstat (limited to 'ethosu/vela/vela.py')
| -rw-r--r-- | ethosu/vela/vela.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index b93774d0..08ab4831 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -245,7 +245,11 @@ def main(args=None): "--show-cpu-operations", action="store_true", help="Show the operations that fall back to the CPU" ) parser.add_argument( - "--keep-scale-placement", action="store_true", help="Keep scale tensors memory placement during scheduling" + "--cache-bias-scale-tensor", + type=ast.literal_eval, + default=True, + choices=[True, False], + help="Controls the caching of the bias & scale tensors in SRAM (default: %(default)s)", ) parser.add_argument( "--cascading", @@ -416,7 +420,7 @@ def main(args=None): use_ifm_streaming=args.ifm_streaming, pareto_metric=args.pareto_metric, use_nhcwb16_between_cascaded_passes=args.nhcwb16_between_cascaded_passes, - keep_scale_placement=args.keep_scale_placement, + cache_bias_scale_tensor=args.cache_bias_scale_tensor, ) model_reader_options = model_reader.ModelReaderOptions() |