batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.primitive.at mean,
mkldnn.primitive.at variance,
mkldnn.primitive.at diff_dst,
mkldnn.memory diff_src) |
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.primitive.at mean,
mkldnn.primitive.at variance,
mkldnn.primitive.at diff_dst,
mkldnn.primitive.at weights_or_workspace,
mkldnn.memory diff_src)
\warning This constructor works for backward_data propagation
- w/ weights but w/o workspace, or
- w/ workspace but w/o weights
|
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.primitive.at mean,
mkldnn.primitive.at variance,
mkldnn.primitive.at diff_dst,
mkldnn.primitive.at weights,
mkldnn.memory diff_src,
mkldnn.memory diff_weights) |
batch_normalization_backward(mkldnn.batch_normalization_backward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.primitive.at mean,
mkldnn.primitive.at variance,
mkldnn.primitive.at diff_dst,
mkldnn.primitive.at weights,
mkldnn.primitive.at workspace,
mkldnn.memory diff_src,
mkldnn.memory diff_weights) |