Home
last modified time | relevance | path

Searched refs:_input_to_output_reduction (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEQLSTMLayer.cpp132 _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()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLQLSTMLayer.cpp101 _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()
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/
H A DCLQLSTMLayer.h321 std::unique_ptr<opencl::kernels::ClGemmLowpMatrixAReductionKernel> _input_to_output_reduction; variable
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
H A DNEQLSTMLayer.h278 std::unique_ptr<cpu::kernels::CpuGemmLowpMatrixAReductionKernel> _input_to_output_reduction; variable