diff options
Diffstat (limited to 'ethosu/vela/vela.py')
| -rw-r--r-- | ethosu/vela/vela.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 1766750e..b1edf34e 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -253,6 +253,13 @@ def main(args=None): choices=[True, False], help="Control if NHCWB16 or NHWC should be used in between cascaded passes (default: %(default)s)", ) + parser.add_argument( + "--softmax-support", + type=ast.literal_eval, + default=False, + choices=[True, False], + help="Control if Softmax should be transformed into a set of npu operations (default: %(default)s)", + ) args = parser.parse_args(args=args) @@ -283,6 +290,7 @@ def main(args=None): block_config_limit=args.block_config_limit, global_memory_clock_scale=args.global_memory_clock_scale, max_blockdep=args.max_block_dependency, + softmax_support=args.softmax_support, ) compiler_options = compiler_driver.CompilerOptions( |