commit 14e8a2041877f277f96fe339f48d4f51e0395a42
author Tim Hall <tim.hall@arm.com> Fri Nov 27 12:23:42 2020 +0000
committer tim.hall <tim.hall@arm.com> Fri Nov 27 14:57:24 2020 +0000
tree 778b399efa01874c8cebece32ff5c9e7b9602739
parent 061eeb4c6a1af7780992ae6d3967905b03e2433a

vela: Rename --keep-scale-placement CLI

 - Changed to --cache-bias-scale-tensor

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I285fe253f03ba98eff36dbe996ad3a57e2ee3d99

ethosu/vela/vela.py

diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index b93774d..08ab483 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -245,7 +245,11 @@
         "--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 @@
         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()