batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.memory dst,
mkldnn.memory mean,
mkldnn.memory variance,
mkldnn.memory workspace)
\warning batch_normalization_forward has 2 constructors with very
similar signatures:
- (pd, src, weights, dst, mean, variance) // 2 in, 3 out
- (pd, src, dst, mean, variance, workspace) // 1 in, 4 out
The only way to distinguish between those is to explicitly
cast all input parameters to their type, i.e.
|
batch_normalization_forward(mkldnn.batch_normalization_forward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src,
mkldnn.primitive.at weights,
mkldnn.memory dst,
mkldnn.memory mean,
mkldnn.memory variance)
\warning batch_normalization_forward has 2 constructors with very
similar signatures:
- (pd, src, weights, dst, mean, variance) // 2 in, 3 out
- (pd, src, dst, mean, variance, workspace) // 1 in, 4 out
The only way to distinguish between those is to explicitly
cast all input parameters to their type, i.e.
|