| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLGEMMLowpReductionKernel.h | 32 struct GEMMLowpReductionKernelInfo;
59 …oid configure(const ICLTensor *input, ICLTensor *output, const GEMMLowpReductionKernelInfo &info) …
71 …pile_context, const ICLTensor *input, ICLTensor *output, const GEMMLowpReductionKernelInfo &info) …
96 …igure(const ICLTensor *mtx_a, ICLTensor *vector_sum_row, const GEMMLowpReductionKernelInfo &info) …
108 …text, const ICLTensor *mtx_a, ICLTensor *vector_sum_row, const GEMMLowpReductionKernelInfo &info) …
121 …st ITensorInfo *mtx_a, const ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info);
145 …igure(const ICLTensor *mtx_b, ICLTensor *vector_sum_col, const GEMMLowpReductionKernelInfo &info) …
157 …text, const ICLTensor *mtx_b, ICLTensor *vector_sum_col, const GEMMLowpReductionKernelInfo &info) …
170 …st ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info);
|
| D | CLGEMMLowpReductionKernel.cpp | 70 …nfigure(const ICLTensor *mtx_a, ICLTensor *vector_sum_row, const GEMMLowpReductionKernelInfo &info) in configure()
75 …ontext, const ICLTensor *mtx_a, ICLTensor *vector_sum_row, const GEMMLowpReductionKernelInfo &info) in configure()
119 …nst ITensorInfo *mtx_a, const ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info) in validate()
151 …nfigure(const ICLTensor *mtx_b, ICLTensor *vector_sum_col, const GEMMLowpReductionKernelInfo &info) in configure()
156 …ontext, const ICLTensor *mtx_b, ICLTensor *vector_sum_col, const GEMMLowpReductionKernelInfo &info) in configure()
191 …nst ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info) in validate()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEGEMMLowpReductionKernel.h | 33 struct GEMMLowpReductionKernelInfo;
62 …virtual void configure(const ITensor *input, ITensor *output, const GEMMLowpReductionKernelInfo &i…
106 …void configure(const ITensor *mtx_a, ITensor *vector_sum_row, const GEMMLowpReductionKernelInfo &i…
119 …st ITensorInfo *mtx_a, const ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info);
167 …void configure(const ITensor *mtx_b, ITensor *vector_sum_col, const GEMMLowpReductionKernelInfo &i…
180 …st ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info);
|
| D | NEGEMMLowpReductionKernel.cpp | 69 …::configure(const ITensor *mtx_a, ITensor *vector_sum_row, const GEMMLowpReductionKernelInfo &info) in configure()
90 …nst ITensorInfo *mtx_a, const ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info) in validate()
186 …::configure(const ITensor *mtx_b, ITensor *vector_sum_col, const GEMMLowpReductionKernelInfo &info) in configure()
211 …nst ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info) in validate()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | KernelDescriptors.h | 153 struct GEMMLowpReductionKernelInfo struct
156 GEMMLowpReductionKernelInfo() = default;
164 GEMMLowpReductionKernelInfo(int32_t k, bool is_reshaped, int32_t scalar, bool mul_by_scalar) in GEMMLowpReductionKernelInfo() function
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEQLSTMLayer.cpp | 231 …>configure(_input_to_input_weights, &_input_to_input_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
232 …re(_recurrent_to_input_weights, &_recurrent_to_input_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
242 …configure(input_to_forget_weights, &_input_to_forget_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
243 …e(recurrent_to_forget_weights, &_recurrent_to_forget_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
244 …on->configure(input_to_cell_weights, &_input_to_cell_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
245 …igure(recurrent_to_cell_weights, &_recurrent_to_cell_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
246 …configure(input_to_output_weights, &_input_to_output_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
247 …e(recurrent_to_output_weights, &_recurrent_to_output_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
251 …duction->configure(_projection_weights, &_projection_eff_bias, GEMMLowpReductionKernelInfo(output_… in configure()
664 …validate(lstm_params.input_to_input_weights(), &eff_bias_info, GEMMLowpReductionKernelInfo(num_uni… in validate()
[all …]
|
| D | NEGEMMLowpMatrixMultiplyCore.cpp | 196 … const GEMMLowpReductionKernelInfo reduction_info(a_to_use->info()->dimension(0), false, 0, false); in configure()
437 const GEMMLowpReductionKernelInfo reduction_info(a_to_use->dimension(0), false, 0, false); in validate()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLQLSTMLayer.cpp | 254 …e_context, _input_to_input_weights, &_input_to_input_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
255 …t, _recurrent_to_input_weights, &_recurrent_to_input_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
257 …_context, input_to_forget_weights, &_input_to_forget_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
258 …, recurrent_to_forget_weights, &_recurrent_to_forget_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
259 …pile_context, input_to_cell_weights, &_input_to_cell_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
260 …text, recurrent_to_cell_weights, &_recurrent_to_cell_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
261 …_context, input_to_output_weights, &_input_to_output_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
262 …, recurrent_to_output_weights, &_recurrent_to_output_eff_bias, GEMMLowpReductionKernelInfo(num_uni… in configure()
265 …e(compile_context, _projection_weights, &_projection_eff_bias, GEMMLowpReductionKernelInfo(output_… in configure()
684 …validate(lstm_params.input_to_input_weights(), &eff_bias_info, GEMMLowpReductionKernelInfo(num_uni… in validate()
[all …]
|
| D | CLGEMMLowpMatrixMultiplyCore.cpp | 185 const GEMMLowpReductionKernelInfo reduction_info {}; in configure()
377 const GEMMLowpReductionKernelInfo reduction_info; in validate()
|