Skip to content

Commit 482c9b2

Browse files
committed
examples: weights layout changed to oihw
1 parent 7264bc8 commit 482c9b2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/cnn_training_f32.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ void simple_net(engine::kind engine_kind) {
349349

350350
auto conv_user_diff_weights_memory
351351
= memory({{conv_weights_tz}, memory::data_type::f32,
352-
memory::format_tag::nhwc},
352+
memory::format_tag::oihw},
353353
eng);
354354
write_to_dnnl_memory(conv_user_diff_weights_buffer.data(),
355355
conv_user_diff_weights_memory);

0 commit comments

Comments
 (0)