Searched refs:_input_to_output_reduction (Results 1 – 4 of 4) sorted by relevance
132 _input_to_output_reduction(), in NEQLSTMLayer()334 …_input_to_output_reduction = std::make_unique<cpu::kernels::CpuGemmLowpMatrixAReductionKernel>… in configure()341 …_input_to_output_reduction->configure(input_to_output_weights->info(), _input_to_output_eff_bias.i… in configure()1307 …NEScheduler::get().schedule_op(_input_to_output_reduction.get(), Window::DimY, _input_to_output_re… in prepare()
101 _input_to_output_reduction(std::make_unique<ClGemmLowpMatrixAReductionKernel>()), in CLQLSTMLayer()263 …_input_to_output_reduction->configure(compile_context, input_to_output_weights->info(), _input_to_… in configure()1172 CLScheduler::get().enqueue_op(*_input_to_output_reduction, input_to_output_red_pack, false); in prepare()
321 std::unique_ptr<opencl::kernels::ClGemmLowpMatrixAReductionKernel> _input_to_output_reduction; variable
278 std::unique_ptr<cpu::kernels::CpuGemmLowpMatrixAReductionKernel> _input_to_output_reduction; variable