Searched refs:_recurrent_to_output_reduction (Results 1 – 4 of 4) sorted by relevance
133 _recurrent_to_output_reduction(), in NEQLSTMLayer()335 …_recurrent_to_output_reduction = std::make_unique<cpu::kernels::CpuGemmLowpMatrixAReductionKernel>… in configure()342 …_recurrent_to_output_reduction->configure(recurrent_to_output_weights->info(), _recurrent_to_outpu… in configure()1314 …NEScheduler::get().schedule_op(_recurrent_to_output_reduction.get(), Window::DimY, _recurrent_to_o… in prepare()
279 std::unique_ptr<cpu::kernels::CpuGemmLowpMatrixAReductionKernel> _recurrent_to_output_reduction; variable
322 … std::unique_ptr<opencl::kernels::ClGemmLowpMatrixAReductionKernel> _recurrent_to_output_reduction; variable
102 _recurrent_to_output_reduction(std::make_unique<ClGemmLowpMatrixAReductionKernel>()), in CLQLSTMLayer()264 …_recurrent_to_output_reduction->configure(compile_context, recurrent_to_output_weights->info(), _r… in configure()1175 … CLScheduler::get().enqueue_op(*_recurrent_to_output_reduction, rec_to_output_red_pack, false); in prepare()