diff --git a/aidge_export_cpp/templates/kernel_forward/_save_outputs.jinja b/aidge_export_cpp/templates/kernel_forward/_save_outputs.jinja index f9962dd1ada9df2951cf4944be36dd604c3abcec..6865be575a613af16fc6a88fd969525abba80d0d 100644 --- a/aidge_export_cpp/templates/kernel_forward/_save_outputs.jinja +++ b/aidge_export_cpp/templates/kernel_forward/_save_outputs.jinja @@ -13,7 +13,7 @@ {{out_name[outidx]|upper}}_STRIDE, #} {{out_name[outidx]}}, {{out_name[outidx]|upper}}_STREAM, - Format::{{out_format[outidx]}}); + Format::NHWC); fclose({{out_name[outidx]|upper}}_STREAM); {% endfor %} #endif