Searched refs:_input_to_input_weights (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/ |
H A D | LSTMParams.h | 41 : _input_to_input_weights(nullptr), in LSTMParams() 84 _input_to_input_weights = input_to_input_weights; in set_cifg_params() 197 return _input_to_input_weights; in input_to_input_weights() 316 const T *_input_to_input_weights;
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A D | CLLSTMLayerQuantized.cpp | 52 …_input_to_input_weights(nullptr), _input_to_forget_weights(nullptr), _input_to_cell_weights(nullpt… in CLLSTMLayerQuantized() 101 _input_to_input_weights = input_to_input_weights; in configure() 549 _input_to_input_weights->mark_as_unused(); in prepare()
|
H A D | CLQLSTMLayer.cpp | 250 _input_to_input_weights = lstm_params.input_to_input_weights(); in configure() 253 …_input_to_input_reduction->configure(compile_context, _input_to_input_weights->info(), _input_to_i… in configure() 397 …const float input_to_input_scale = _input_to_input_weights->info()->quantization_info().unifo… in configure() 1139 …ITensorPack input_to_input_red_pack = { { ACL_SRC, _input_to_input_weights }, { ACL_DST, &_input_t… in prepare() 1149 _input_to_input_weights->mark_as_unused(); in prepare()
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
H A D | NELSTMLayerQuantized.cpp | 51 …ll_tensor(), _slice_output_tensor(), _dequantize(), _quantize(), _input_to_input_weights(nullptr),… in NELSTMLayerQuantized() 89 _input_to_input_weights = input_to_input_weights; in configure() 516 _input_to_input_weights->mark_as_unused(); in prepare()
|
H A D | NEQLSTMLayer.cpp | 321 _input_to_input_weights = lstm_params.input_to_input_weights(); in configure() 326 …_input_to_input_reduction->configure(_input_to_input_weights->info(), _input_to_input_eff_bias.inf… in configure() 473 …const float input_to_input_scale = _input_to_input_weights->info()->quantization_info().unifo… in configure() 1248 { TensorType::ACL_SRC, _input_to_input_weights }, in prepare() 1264 _input_to_input_weights->mark_as_unused(); in prepare()
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
H A D | NELSTMLayerQuantized.h | 176 const ITensor *_input_to_input_weights; variable
|
H A D | NEQLSTMLayer.h | 338 const ITensor *_input_to_input_weights
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
H A D | CLLSTMLayerQuantized.h | 199 const ICLTensor *_input_to_input_weights; variable
|
H A D | CLQLSTMLayer.h | 379 const ICLTensor *_input_to_input_weights
|