Searched refs:batch_normalization_layer (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
H A D | BatchNormalizationLayer.cpp | 40 SimpleTensor<T> batch_normalization_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &mean, … in batch_normalization_layer() function 77 template SimpleTensor<float> batch_normalization_layer(const SimpleTensor<float> &src, const Simple… 79 template SimpleTensor<half> batch_normalization_layer(const SimpleTensor<half> &src, const SimpleTe…
|
H A D | BatchNormalizationLayer.h | 39 SimpleTensor<T> batch_normalization_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &mean, …
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | BatchNormalizationLayerFixture.h | 157 …return reference::batch_normalization_layer(ref_src, ref_mean, ref_var, ref_beta, ref_gamma, epsil… in compute_reference()
|
H A D | BatchNormalizationLayerFusionFixture.h | 175 …return reference::batch_normalization_layer(conv_res, bn_mean, bn_var, bn_beta, bn_gamma, epsilon,… in compute_reference()
|