Home
last modified time | relevance | path

Searched defs:var_vec (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/batchnormalization/impl/SVE/
H A Dfp32.cpp73 const auto var_vec = svld1_f32(pg, input_var + x); in fp32_sve_batch_normalization() local
H A Dfp16.cpp73 const auto var_vec = svld1_f16(pg, input_var + x); in fp16_sve_batch_normalization() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/
H A Dimpl.cpp61 auto var_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_conv() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/
H A Dimpl.cpp61 auto var_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nhwc() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/batchnormalization/impl/NEON/
H A Dfp16.cpp79 const auto var_vec = wrapper::vloadq(input_var + x); in batch_normalization() local
H A Dfp32.cpp78 const auto var_vec = wrapper::vloadq(input_var + x); in batch_normalization() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/
H A Dall.cpp62 auto var_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nchw() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEBatchNormalizationLayerKernel.cpp184 auto var_vec = wrapper::vdup_n(var, ExactTagType{}); in batch_normalization_nchw() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDistributionTemplates.h178 const Vec var_vec = Vec(std); in normal_fill_VSX() local