Searched refs:_input_modulation_gate_output (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.cpp | 55 …t_gate_output(), _input_gate_output(), _output_gate_output(), _input_modulation_gate_output(), _ce… in CLLSTMLayerQuantized()
232 _memory_group.manage(&_input_modulation_gate_output); in configure()
233 …_input_modulation_gate_output.allocator()->init(TensorInfo(_input_modulation_gate_input.info()->te… in configure()
234 …te.configure(compile_context, &_input_modulation_gate_input, &_input_modulation_gate_output, Activ… in configure()
251 …put_mod_gate.configure(compile_context, &_input_gate_output, &_input_modulation_gate_output, &_cel… in configure()
252 _input_modulation_gate_output.allocator()->allocate(); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.cpp | 55 …_input_gate_output(), _output_gate_output(), _input_modulation_gate_output(), _cell_state1(), _cel… in NELSTMLayerQuantized()
200 _memory_group.manage(&_input_modulation_gate_output); in configure()
201 …_input_modulation_gate_output.allocator()->init(TensorInfo(_input_modulation_gate_input.info()->te… in configure()
202 …_tanh_modulation_gate.configure(&_input_modulation_gate_input, &_input_modulation_gate_output, Act… in configure()
219 …_mul2.configure(&_input_gate_output, &_input_modulation_gate_output, &_cell_state2, 1, ConvertPoli… in configure()
220 _input_modulation_gate_output.allocator()->allocate(); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.h | 205 Tensor _input_modulation_gate_output; variable
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.h | 228 CLTensor _input_modulation_gate_output; variable
|