We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7264bc8 commit 482c9b2Copy full SHA for 482c9b2
1 file changed
examples/cnn_training_f32.cpp
@@ -349,7 +349,7 @@ void simple_net(engine::kind engine_kind) {
349
350
auto conv_user_diff_weights_memory
351
= memory({{conv_weights_tz}, memory::data_type::f32,
352
- memory::format_tag::nhwc},
+ memory::format_tag::oihw},
353
eng);
354
write_to_dnnl_memory(conv_user_diff_weights_buffer.data(),
355
conv_user_diff_weights_memory);
0 commit comments