Searched refs:_recurrent_to_input_weights (Results 1 – 9 of 9) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/common/ |
| D | LSTMParams.h | 42 _recurrent_to_input_weights(nullptr), in LSTMParams()
85 _recurrent_to_input_weights = recurrent_to_input_weights; in set_cifg_params()
202 return _recurrent_to_input_weights; in recurrent_to_input_weights()
317 const T *_recurrent_to_input_weights; variable
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.h | 180 const ITensor *_recurrent_to_input_weights; variable
|
| D | NEQLSTMLayer.h | 342 const ITensor *_recurrent_to_input_weights{ nullptr };
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.h | 203 const ICLTensor *_recurrent_to_input_weights; variable
|
| D | CLQLSTMLayer.h | 383 const ICLTensor *_recurrent_to_input_weights{ nullptr };
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.cpp | 52 …t_to_cell_weights(nullptr), _input_to_output_weights(nullptr), _recurrent_to_input_weights(nullptr… in NELSTMLayerQuantized()
93 _recurrent_to_input_weights = recurrent_to_input_weights; in configure()
523 _recurrent_to_input_weights->mark_as_unused(); in prepare()
|
| D | NEQLSTMLayer.cpp | 322 _recurrent_to_input_weights = lstm_params.recurrent_to_input_weights(); in configure()
327 …_recurrent_to_input_reduction->configure(_recurrent_to_input_weights->info(), _recurrent_to_input_… in configure()
479 …const float recurrent_to_input_scale = _recurrent_to_input_weights->info()->quantization_info().un… in configure()
1255 { TensorType::ACL_SRC, _recurrent_to_input_weights }, in prepare()
1265 _recurrent_to_input_weights->mark_as_unused(); in prepare()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.cpp | 52 …t_to_cell_weights(nullptr), _input_to_output_weights(nullptr), _recurrent_to_input_weights(nullptr… in CLLSTMLayerQuantized()
105 _recurrent_to_input_weights = recurrent_to_input_weights; in configure()
556 _recurrent_to_input_weights->mark_as_unused(); in prepare()
|
| D | CLQLSTMLayer.cpp | 251 _recurrent_to_input_weights = lstm_params.recurrent_to_input_weights(); in configure()
254 …_recurrent_to_input_reduction->configure(compile_context, _recurrent_to_input_weights->info(), _re… in configure()
403 …const float recurrent_to_input_scale = _recurrent_to_input_weights->info()->quantization_info().un… in configure()
1142 …ITensorPack rec_to_input_red_pack = { { ACL_SRC, _recurrent_to_input_weights }, { ACL_DST, &_recur… in prepare()
1150 _recurrent_to_input_weights->mark_as_unused(); in prepare()
|