Home
last modified time | relevance | path

Searched refs:NEBatchNormalizationLayerKernel (Results 1 – 8 of 8) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/
DNEBatchNormalizationLayerKernel.h36 class NEBatchNormalizationLayerKernel : public INEKernel
44 NEBatchNormalizationLayerKernel();
46 NEBatchNormalizationLayerKernel(const NEBatchNormalizationLayerKernel &) = delete;
48 NEBatchNormalizationLayerKernel &operator=(const NEBatchNormalizationLayerKernel &) = delete;
50 NEBatchNormalizationLayerKernel(NEBatchNormalizationLayerKernel &&) = default;
52 NEBatchNormalizationLayerKernel &operator=(NEBatchNormalizationLayerKernel &&) = default;
54 ~NEBatchNormalizationLayerKernel() = default;
125 using BatchNormFunctionPtr = void (NEBatchNormalizationLayerKernel::*)(const Window &window);
DNEBatchNormalizationLayerKernel.cpp151 void NEBatchNormalizationLayerKernel::batch_normalization_nchw(const Window &window) in batch_normalization_nchw()
256 void NEBatchNormalizationLayerKernel::configure_non_fused() in configure_non_fused()
262 …_func = &NEBatchNormalizationLayerKernel::batch_normalization_nchw<float16_t, false, detail::dummy… in configure_non_fused()
266 …_func = &NEBatchNormalizationLayerKernel::batch_normalization_nchw<float, false, detail::dummy<flo… in configure_non_fused()
274 void NEBatchNormalizationLayerKernel::configure_fused() in configure_fused()
279 …{ ActivationLayerInfo::ActivationFunction::RELU, &NEBatchNormalizationLayerKernel::batch_normaliza… in configure_fused()
280 …{ ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, &NEBatchNormalizationLayerKernel::batch_n… in configure_fused()
281 …{ ActivationLayerInfo::ActivationFunction::LU_BOUNDED_RELU, &NEBatchNormalizationLayerKernel::batc… in configure_fused()
287 …{ ActivationLayerInfo::ActivationFunction::RELU, &NEBatchNormalizationLayerKernel::batch_normaliza… in configure_fused()
288 …{ ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, &NEBatchNormalizationLayerKernel::batch_n… in configure_fused()
[all …]
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEBatchNormalizationLayer.h36 class NEBatchNormalizationLayerKernel; variable
110 …std::unique_ptr<NEBatchNormalizationLayerKernel> _norm_kernel; /**< Batch normalization layer kern…
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEBatchNormalizationLayer.cpp49 _norm_kernel = std::make_unique<NEBatchNormalizationLayerKernel>(); in configure()
56 …ARM_COMPUTE_RETURN_ON_ERROR(NEBatchNormalizationLayerKernel::validate(input, output, mean, var, be… in validate()
/external/ComputeLibrary/docs/user_guide/
Drelease_version_and_change_log.dox357 - @ref NEBatchNormalizationLayerKernel
727 - @ref NEBatchNormalizationLayerKernel
1510 - @ref NEBatchNormalizationLayerKernel / @ref NEBatchNormalizationLayer
/external/ComputeLibrary/src/
DCMakeLists.txt375 core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp
DBUILD.bazel383 "core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp",
/external/ComputeLibrary/
DAndroid.bp172 "src/core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp",