25 #ifndef ACL_ARM_COMPUTE_GRAPH_BACKENDS_FUSEDCONVOLUTIONBATCHNORMALIZATIONFUNCTION_H
26 #define ACL_ARM_COMPUTE_GRAPH_BACKENDS_FUSEDCONVOLUTIONBATCHNORMALIZATIONFUNCTION_H
39 template <
typename TargetInfo,
typename FusedLayerTypes>
47 : _conv_layer(memory_manager), _fused_batch_norm_layer(), _fused_bias(), _is_prepared(false)
95 _fused_batch_norm_layer.configure(weights, mean, var,
nullptr,
nullptr,
bias, beta, gamma,
epsilon);
100 _fused_batch_norm_layer.configure(weights, mean, var,
nullptr, &_fused_bias,
nullptr, beta, gamma,
epsilon);
101 bias_to_use = &_fused_bias;
109 _fused_bias.allocator()->allocate();
124 _fused_batch_norm_layer.run();
130 typename FusedLayerTypes::ConvolutionLayer _conv_layer;
131 typename FusedLayerTypes::FuseBatchNormalization _fused_batch_norm_layer;
139 #endif // ACL_ARM_COMPUTE_GRAPH_BACKENDS_FUSEDCONVOLUTIONBATCHNORMALIZATIONFUNCTION_H