Searched refs:FuseBatchNormalizationType (Results 1 – 12 of 12) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLFuseBatchNormalization.h | 83 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
102 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
123 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEFuseBatchNormalizationKernel.cpp | 50 FuseBatchNormalizationType fbn_type;
71 …ta.dl == DataLayout::NHWC && data.isa.fp16 && data.fbn_type == FuseBatchNormalizationType::CONVOLU… in __anon111380ba0202()
79 …ta.dl == DataLayout::NCHW && data.isa.fp16 && data.fbn_type == FuseBatchNormalizationType::CONVOLU… in __anon111380ba0302()
87 …ta.dl == DataLayout::NHWC && data.isa.fp16 && data.fbn_type == FuseBatchNormalizationType::DEPTHWI… in __anon111380ba0402()
95 …ta.dl == DataLayout::NCHW && data.isa.fp16 && data.fbn_type == FuseBatchNormalizationType::DEPTHWI… in __anon111380ba0502()
103 …taType::F32 && data.dl == DataLayout::NHWC && data.fbn_type == FuseBatchNormalizationType::CONVOLU… in __anon111380ba0602()
111 …taType::F32 && data.dl == DataLayout::NCHW && data.fbn_type == FuseBatchNormalizationType::CONVOLU… in __anon111380ba0702()
119 …taType::F32 && data.dl == DataLayout::NHWC && data.fbn_type == FuseBatchNormalizationType::DEPTHWI… in __anon111380ba0802()
127 …taType::F32 && data.dl == DataLayout::NCHW && data.fbn_type == FuseBatchNormalizationType::DEPTHWI… in __anon111380ba0902()
156 float epsilon, FuseBatchNormalizationType fbn_type) in validate_arguments()
[all …]
|
| D | NEFuseBatchNormalizationKernel.h | 71 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
92 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLFuseBatchNormalizationKernel.h | 67 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
86 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
107 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
|
| D | CLFuseBatchNormalizationKernel.cpp | 45 float epsilon, FuseBatchNormalizationType fbn_type) in validate_arguments()
56 if(fbn_type == FuseBatchNormalizationType::CONVOLUTION) in validate_arguments()
112 float epsilon, FuseBatchNormalizationType fbn_type) in configure()
120 float epsilon, FuseBatchNormalizationType fbn_type) in configure()
167 …build_opts.add_option_if(fbn_type == FuseBatchNormalizationType::CONVOLUTION, "-DDIM2=" + support:… in configure()
185 float epsilon, FuseBatchNormalizationType fbn_type) in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEFuseBatchNormalization.h | 80 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
101 …float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUT…
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLFuseBatchNormalization.cpp | 47 float epsilon, FuseBatchNormalizationType fbn_type) in configure()
55 float epsilon, FuseBatchNormalizationType fbn_type) in configure()
64 float epsilon, FuseBatchNormalizationType fbn_type) in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEFuseBatchNormalization.cpp | 46 float epsilon, FuseBatchNormalizationType fbn_type) in configure()
58 float epsilon, FuseBatchNormalizationType fbn_type) in validate()
|
| /external/ComputeLibrary/arm_compute/graph/backends/ |
| D | FusedDepthwiseConvolutionBatchNormalizationFunction.h | 88 …ghts, mean, var, nullptr, nullptr, bias, beta, gamma, epsilon, FuseBatchNormalizationType::DEPTHWI… in configure()
93 …an, var, nullptr, &_fused_bias, nullptr, beta, gamma, epsilon, FuseBatchNormalizationType::DEPTHWI… in configure()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | FuseBatchNormalizationFixture.h | 92 const FuseBatchNormalizationType fuse_bn_type = dims_weights == 3 ? in compute_target()
93 … FuseBatchNormalizationType::DEPTHWISECONVOLUTION : in compute_target()
94 FuseBatchNormalizationType::CONVOLUTION; in compute_target()
|
| /external/ComputeLibrary/utils/ |
| D | TypePrinter.h | 3053 inline ::std::ostream &operator<<(::std::ostream &os, const FuseBatchNormalizationType &fuse_type)
3057 case FuseBatchNormalizationType::CONVOLUTION:
3060 case FuseBatchNormalizationType::DEPTHWISECONVOLUTION:
3075 inline std::string to_string(const FuseBatchNormalizationType &fuse_type) in to_string()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | Types.h | 160 enum class FuseBatchNormalizationType enum
|