Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/
H A Dall.cpp98 … auto dwc_bias_tmp_scalar = (dwc_bias_in_scalar - mean) / std::sqrt(var + ScalarType(epsilon)); in fused_batch_normalization_dwc_nchw() local
99 dwc_bias_out[id[2]] = (dwc_bias_tmp_scalar * gamma) + beta; in fused_batch_normalization_dwc_nchw()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/
H A Dimpl.cpp136 … auto dwc_bias_tmp_scalar = (dwc_bias_in_scalar - mean) / std::sqrt(var + ScalarType(epsilon)); in fused_batch_normalization_dwc_nhwc() local
137 dwc_bias_out[x] = (dwc_bias_tmp_scalar * gamma) + beta; in fused_batch_normalization_dwc_nhwc()