Searched refs:fused_batch_normalization_dwc_nhwc (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/ |
H A D | impl.cpp | 31 void fused_batch_normalization_dwc_nhwc(const ITensor *dwc_weights, const ITensor *dwc_bias, ITenso… in fused_batch_normalization_dwc_nhwc() function 149 template void fused_batch_normalization_dwc_nhwc<float32_t>(const ITensor *dwc_weights, const ITens… 153 template void fused_batch_normalization_dwc_nhwc<float16_t>(const ITensor *dwc_weights, const ITens…
|
H A D | impl.h | 35 void fused_batch_normalization_dwc_nhwc(const ITensor *dwc_weights, const ITensor *dwc_bias, ITenso…
|
H A D | fp32.cpp | 35 …return fused_batch_normalization_dwc_nhwc<float32_t>(dwc_weights, dwc_bias, fused_weights, fused_b… in fused_batch_normalization_dwc_nhwc_f32()
|
H A D | fp16.cpp | 36 …return fused_batch_normalization_dwc_nhwc<float16_t>(dwc_weights, dwc_bias, fused_weights, fused_b… in fused_batch_normalization_dwc_nhwc_f16()
|