aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/vela.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r--ethosu/vela/vela.py8
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()