[GPU] Minor fix for dynamic mobilebert (#15909)

Signed-off-by: Andrew Park <andrew.park@intel.com>
This commit is contained in:
Andrew Kwangwoong Park 2023-02-26 13:22:44 +09:00 committed by GitHub
parent fabf67ee5e
commit 39e63ace67
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1693,7 +1693,7 @@ format layout_optimizer::get_preferred_format(program_node& node) {
}
// In case of input -> ... -> quantize -> concat
if (expected == format::any
if (layout.is_static() && expected == format::any
&& (node.get_users().size() == 1 && node.get_users().front()->is_type<concatenation>())
&& (layout.batch() < 4 && layout.feature() < 4)) {
expected = format::get_default_format(layout.get_rank(), false, false);