batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc,
primitive.at src,
primitive.at mean,
primitive.at variance,
primitive.at diff_dst,
memory diff_src) |
batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc,
primitive.at src,
primitive.at mean,
primitive.at variance,
primitive.at diff_dst,
primitive.at weights_or_workspace,
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(batch_normalization_backward.primitive_desc aprimitive_desc,
primitive.at src,
primitive.at mean,
primitive.at variance,
primitive.at diff_dst,
primitive.at weights,
memory diff_src,
memory diff_weights) |
batch_normalization_backward(batch_normalization_backward.primitive_desc aprimitive_desc,
primitive.at src,
primitive.at mean,
primitive.at variance,
primitive.at diff_dst,
primitive.at weights,
primitive.at workspace,
memory diff_src,
memory diff_weights) |