Searched refs:NEFuseBatchNormalization (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEFuseBatchNormalization.h | 38 class NEFuseBatchNormalization : public IFunction
42 NEFuseBatchNormalization();
44 NEFuseBatchNormalization(const NEFuseBatchNormalization &) = delete;
46 NEFuseBatchNormalization &operator=(const NEFuseBatchNormalization &) = delete;
48 NEFuseBatchNormalization(NEFuseBatchNormalization &&) = default;
50 NEFuseBatchNormalization &operator=(NEFuseBatchNormalization &&) = default;
52 ~NEFuseBatchNormalization();
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEFuseBatchNormalization.cpp | 36 NEFuseBatchNormalization::~NEFuseBatchNormalization() = default;
38 NEFuseBatchNormalization::NEFuseBatchNormalization() in NEFuseBatchNormalization() function in arm_compute::NEFuseBatchNormalization
43 void NEFuseBatchNormalization::configure(const ITensor *input_weights, const ITensor *bn_mean, cons… in configure()
52 Status NEFuseBatchNormalization::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_m… in validate()
60 void NEFuseBatchNormalization::run() in run()
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | FuseBatchNormalization.cpp | 48 …ationConvFixture = FuseBatchNormalizationFixture<Tensor, Accessor, NEFuseBatchNormalization, 4, T>;
50 …zationDWCFixture = FuseBatchNormalizationFixture<Tensor, Accessor, NEFuseBatchNormalization, 3, T>;
|
| D | BatchNormalizationLayer.cpp | 172 …ionLayerFusionValidationFixture<Tensor, Accessor, NEConvolutionLayer, NEFuseBatchNormalization, T>;
198 bool has_error = bool(NEFuseBatchNormalization::validate(
|
| /external/ComputeLibrary/src/graph/backends/NEON/ |
| D | NEFunctionFactory.cpp | 87 using FuseBatchNormalization = NEFuseBatchNormalization;
|
| /external/ComputeLibrary/docs/ |
| D | 06_functions_list.dox | 140 - @ref NEFuseBatchNormalization
|
| D | 00_introduction.dox | 796 - @ref NEFuseBatchNormalizationKernel / @ref NEFuseBatchNormalization
|
| /external/ComputeLibrary/ |
| D | Android.bp | 738 "src/runtime/NEON/functions/NEFuseBatchNormalization.cpp",
|