Searched defs:fused_bias (Results 1 – 11 of 11) sorted by relevance
/external/ComputeLibrary/src/core/CL/kernels/ |
D | CLFuseBatchNormalizationKernel.cpp | 43 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/ |
D | CLFuseBatchNormalization.cpp | 45 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/ |
D | NEFuseBatchNormalization.cpp | 44 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/ |
D | all.cpp | 32 …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/ |
D | NEFuseBatchNormalizationKernel.cpp | 154 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/ |
D | fp16.cpp | 32 …const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_conv_f16()
|
D | fp32.cpp | 31 …const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_conv_f32()
|
D | impl.cpp | 31 …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/ |
D | fp32.cpp | 32 …2(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc_f32()
|
D | fp16.cpp | 33 …6(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc_f16()
|
D | impl.cpp | 31 …c(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *fused_bias, in fused_batch_normalization_dwc_nhwc()
|