[MLBEDSW-2335] SoftMax int16
Added graph rewrite of Softmax for int16.
Change-Id: Id7885af6056a23e8b8362fb61ae94283251eb398
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 1766750..b1edf34 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -253,6 +253,13 @@
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 @@
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(