Skip to content

Commit

Permalink
[GPU] Changing format mapping for lstm_weights_dio
Browse files Browse the repository at this point in the history
  • Loading branch information
Lyamin-Roman committed Jun 15, 2023
1 parent a14cd3a commit 7f0d050
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -659,7 +659,7 @@ kernel_selector::weights_layout to_weights_layout(format f, bool is_grouped) {
case format::g_os_y_is_x_osv8_isv4:
return kernel_selector::weights_layout::g_os_y_is_x_osv8_isv4;
case format::lstm_weights_dio:
return kernel_selector::weights_layout::os_iyx_osv16_rotate_180;
return kernel_selector::weights_layout::dlstm_dir_io;
default:
throw std::invalid_argument("Unable to convert tensor layout " + fmt_to_str(f) + " to weights layout");
}
Expand Down

0 comments on commit 7f0d050

Please sign in to comment.