Home
last modified time | relevance | path

Searched refs:_normalize_kernel (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/runtime/NEON/functions/
DNEL2NormalizeLayer.cpp41 : _memory_group(std::move(memory_manager)), _reduce_func(), _normalize_kernel(), _sumsq() in NEL2NormalizeLayer()
53 _normalize_kernel = arm_compute::support::cpp14::make_unique<NEL2NormalizeLayerKernel>(); in configure()
54 _normalize_kernel->configure(input, &_sumsq, output, axis, epsilon); in configure()
86 NEScheduler::get().schedule(_normalize_kernel.get(), Window::DimY); in run()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLL2NormalizeLayer.cpp47 _normalize_kernel(support::cpp14::make_unique<CLL2NormalizeLayerKernel>()), in CLL2NormalizeLayer()
70 _normalize_kernel->configure(compile_context, input, &_sumsq, output, axis, epsilon); in configure()
102 CLScheduler::get().enqueue(*_normalize_kernel, true); in run()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEL2NormalizeLayer.h87 std::unique_ptr<NEL2NormalizeLayerKernel> _normalize_kernel; variable
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLL2NormalizeLayer.h101 std::unique_ptr<CLL2NormalizeLayerKernel> _normalize_kernel; variable