Home
last modified time | relevance | path

Searched defs:fused_bias (Results 1 – 11 of 11) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLFuseBatchNormalizationKernel.cpp43 const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate_arguments()
110 ICLTensor *fused_weights, ICLTensor *fused_bias, in configure()
118 ICLTensor *fused_weights, ICLTensor *fused_bias, in configure()
183 … const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFuseBatchNormalization.cpp45 ICLTensor *fused_weights, ICLTensor *fused_bias, in configure()
53 ICLTensor *fused_weights, ICLTensor *fused_bias, in configure()
62 … const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFuseBatchNormalization.cpp44 ITensor *fused_weights, ITensor *fused_bias, in configure()
56 … const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate()
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/
Dall.cpp32 …w(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nchw()
128 …2(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nchw_f32()
136 …6(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nchw_f16()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEFuseBatchNormalizationKernel.cpp154 const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate_arguments()
221 ITensor *fused_weights, ITensor *fused_bias, in configure()
272 … const ITensorInfo *fused_weights, const ITensorInfo *fused_bias, in validate()
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/
Dfp16.cpp32 …const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_conv_f16()
Dfp32.cpp31 …const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_conv_f32()
Dimpl.cpp31 …const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_conv()
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/
Dfp32.cpp32 …2(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc_f32()
Dfp16.cpp33 …6(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc_f16()
Dimpl.cpp31 …c(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc()