Go to the source code of this file.
|
template<typename T , bool fused_activation, typename F > |
void | batch_normalization_nchw (const Window &window, ITensor *in, ITensor *out, const ITensor *in_mean, const ITensor *in_var, const ITensor *in_beta, const ITensor *in_gamma, float epsilon, ActivationLayerInfo act_info) |
|
template<typename T > |
void | fused_batch_normalization_conv (const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, float epsilon, const Window &window) |
|
template<typename T > |
void | fused_batch_normalization_dwc_nchw (const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, float epsilon, const Window &window) |
|