diff options
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
| -rw-r--r-- | ethosu/vela/compiler_driver.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index 7b1ea213..d17f1e5b 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -174,7 +174,7 @@ def compiler_driver(nng, arch, options, scheduler_options): # block config, and calc and pack the scales and biases weight_compressor.update_pass_weight_and_scale_tensors(nng, arch) - if not scheduler_options.keep_scale_placement: + if scheduler_options.cache_bias_scale_tensor: scheduler.move_scales_to_fast_storage(nng, arch) # LiveRanges for constant tensors for all Npu subgraphs |