Searched defs:fused_weights (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 …on_dwc_nchw(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_dwc_nchw()
128 …wc_nchw_f32(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_dwc_nchw_f32()
136 …wc_nchw_f16(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… 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 …_conv_f16(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_conv_f16()
|
| D | fp32.cpp | 31 …_conv_f32(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_conv_f32()
|
| D | impl.cpp | 31 …tion_conv(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_conv()
|
| /external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/ |
| D | fp32.cpp | 32 …wc_nhwc_f32(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_dwc_nhwc_f32()
|
| D | fp16.cpp | 33 …wc_nhwc_f16(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_dwc_nhwc_f16()
|
| D | impl.cpp | 31 …on_dwc_nhwc(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITensor *… in fused_batch_normalization_dwc_nhwc()
|