Skip to content

Commit ccf914f

Browse files
committed
examples: fix typos
1 parent 09d1d87 commit ccf914f

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

examples/simple_net.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,8 @@ void simple_net(){
168168
auto pool_indices_memory = memory(pool_pd.workspace_primitive_desc());
169169

170170
/* create pooling primitive an add it to net */
171-
net.push_back(pooling_forward(pool_pd, lrn_dst_memory, pool_indices_memory,
172-
pool_dst_memory));
171+
net.push_back(pooling_forward(pool_pd, lrn_dst_memory, pool_dst_memory,
172+
pool_indices_memory));
173173

174174
/* create reorder between internal and user data if it is needed and
175175
* add it to net after pooling */

examples/simple_training_net.c

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -582,14 +582,15 @@ mkldnn_status_t simple_net()
582582
mkldnn_primitive_at_t lrn_diff_dsts[]
583583
= { mkldnn_primitive_at(relu_dst_memory,
584584
0), // lrn_bwd requires src as first input
585-
mkldnn_primitive_at(pool_diff_dst_memory, 0),
585+
mkldnn_primitive_at(pool_diff_src_memory, 0),
586586
mkldnn_primitive_at(lrn_workspace_memory, 0) };
587587

588588
const_mkldnn_primitive_t lrn_diff_srcs[] = { lrn_diff_src_memory };
589589

590590
/* finally create backward lrn primitive */
591591
mkldnn_primitive_t lrn_bwd;
592-
mkldnn_primitive_create(&lrn_bwd, lrn_bwd_pd, lrn_diff_dsts, lrn_diff_srcs);
592+
CHECK(mkldnn_primitive_create(&lrn_bwd, lrn_bwd_pd, lrn_diff_dsts,
593+
lrn_diff_srcs));
593594

594595
/* Backward relu */
595596
const mkldnn_memory_desc_t *relu_diff_dst_md

0 commit comments

Comments
 (0)