24.02.1
|
#include "src/cpu/kernels/fuse_batch_normalization/generic/impl.h"
#include "src/cpu/kernels/fuse_batch_normalization/nhwc/neon/impl.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::cpu | |
Functions | |
void | fused_batch_normalization_dwc_nhwc_f32 (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) |