Home
last modified time | relevance | path

Searched refs:dwc_bias_in_scalar (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/
H A Dall.cpp72 auto dwc_bias_in_scalar = ScalarType(0.0); in fused_batch_normalization_dwc_nchw() local
95 dwc_bias_in_scalar = dwc_bias_in[id[2]]; in fused_batch_normalization_dwc_nchw()
98 … auto dwc_bias_tmp_scalar = (dwc_bias_in_scalar - mean) / std::sqrt(var + ScalarType(epsilon)); in fused_batch_normalization_dwc_nchw()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/
H A Dimpl.cpp70 auto dwc_bias_in_scalar = ScalarType(0); in fused_batch_normalization_dwc_nhwc() local
133 dwc_bias_in_scalar = dwc_bias_in[x]; in fused_batch_normalization_dwc_nhwc()
136 … auto dwc_bias_tmp_scalar = (dwc_bias_in_scalar - mean) / std::sqrt(var + ScalarType(epsilon)); in fused_batch_normalization_dwc_nhwc()