Searched refs:_recurrent_to_input_reduction (Results 1 – 4 of 4) sorted by relevance
127 _recurrent_to_input_reduction(), in NEQLSTMLayer()325 …_recurrent_to_input_reduction = std::make_unique<cpu::kernels::CpuGemmLowpMatrixAReductionKernel>(… in configure()327 …_recurrent_to_input_reduction->configure(_recurrent_to_input_weights->info(), _recurrent_to_input_… in configure()1258 …NEScheduler::get().schedule_op(_recurrent_to_input_reduction.get(), Window::DimY, _recurrent_to_in… in prepare()
96 _recurrent_to_input_reduction(std::make_unique<ClGemmLowpMatrixAReductionKernel>()), in CLQLSTMLayer()254 …_recurrent_to_input_reduction->configure(compile_context, _recurrent_to_input_weights->info(), _re… in configure()1143 … CLScheduler::get().enqueue_op(*_recurrent_to_input_reduction, rec_to_input_red_pack, false); in prepare()
316 … std::unique_ptr<opencl::kernels::ClGemmLowpMatrixAReductionKernel> _recurrent_to_input_reduction; variable
273 std::unique_ptr<cpu::kernels::CpuGemmLowpMatrixAReductionKernel> _recurrent_to_input_reduction; variable