Searched refs:_normalize_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEL2NormalizeLayer.cpp | 41 : _memory_group(std::move(memory_manager)), _reduce_func(), _normalize_kernel(), _sumsq() in NEL2NormalizeLayer()
55 _normalize_kernel = std::make_unique<NEL2NormalizeLayerKernel>(); in configure()
56 _normalize_kernel->configure(input, &_sumsq, output, axis, epsilon); in configure()
88 NEScheduler::get().schedule(_normalize_kernel.get(), Window::DimY); in run()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLL2NormalizeLayer.cpp | 48 _normalize_kernel(std::make_unique<CLL2NormalizeLayerKernel>()), in CLL2NormalizeLayer()
73 _normalize_kernel->configure(compile_context, input, &_sumsq, output, axis, epsilon); in configure()
105 CLScheduler::get().enqueue(*_normalize_kernel, true); in run()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| H A D | NEL2NormalizeLayer.h | 97 std::unique_ptr<NEL2NormalizeLayerKernel> _normalize_kernel; variable
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| H A D | CLL2NormalizeLayer.h | 111 std::unique_ptr<CLL2NormalizeLayerKernel> _normalize_kernel; variable
|