D | NENormalizationLayerKernel.cpp | 67 NENormalizationLayerKernel::NENormalizationLayerKernel() in NENormalizationLayerKernel() function in arm_compute::NENormalizationLayerKernel 72 void NENormalizationLayerKernel::configure(const ITensor *input, const ITensor *input_squared, ITen… in configure() 98 _func = &NENormalizationLayerKernel::normalize_float<float, 4, 0, true>; in configure() 102 _func = &NENormalizationLayerKernel::normalize_float<float, 4, 0, false>; in configure() 109 _func = &NENormalizationLayerKernel::normalize_float<float, 4, 1, true>; in configure() 113 _func = &NENormalizationLayerKernel::normalize_float<float, 4, 1, false>; in configure() 117 _func = &NENormalizationLayerKernel::normalize_float<float, 4, 2, false>; in configure() 133 _func = &NENormalizationLayerKernel::normalize_float<float16_t, 8, 0, true>; in configure() 137 … _func = &NENormalizationLayerKernel::normalize_float<float16_t, 8, 0, false>; in configure() 144 _func = &NENormalizationLayerKernel::normalize_float<float16_t, 8, 1, true>; in configure() [all …]
|