Searched refs:_normalization_kernel (Results 1 – 2 of 2) sorted by relevance
37 …: _memory_group(std::move(memory_manager)), _normalization_kernel(), _is_nchw(false), _permute_inp… in NEInstanceNormalizationLayer()51 _normalization_kernel = std::make_unique<NEInstanceNormalizationLayerKernel>(); in configure()62 _normalization_kernel->configure(&_permuted_input, &_permuted_output, kernel_descriptor); in configure()71 _normalization_kernel->configure(input, output, kernel_descriptor); in configure()92 NEScheduler::get().schedule(_normalization_kernel.get(), Window::DimZ); in run()
99 std::unique_ptr<NEInstanceNormalizationLayerKernel> _normalization_kernel; variable