Searched refs:_input_modulation_gate_input (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.cpp | 55 …_output_gate_input(), _input_modulation_gate_input(), _forget_gate_output(), _input_gate_output(),… in CLLSTMLayerQuantized()
200 _memory_group.manage(&_input_modulation_gate_input); in configure()
201 …_slice_cell_tensor.configure(compile_context, &_output_lowp, &_input_modulation_gate_input, { 2 * … in configure()
212 _memory_group.manage(&_input_modulation_gate_input); in configure()
213 …_slice_cell_tensor.configure(compile_context, &_output_lowp, &_input_modulation_gate_input, { 2 * … in configure()
233 …_input_modulation_gate_output.allocator()->init(TensorInfo(_input_modulation_gate_input.info()->te… in configure()
234 …_tanh_modulation_gate.configure(compile_context, &_input_modulation_gate_input, &_input_modulation… in configure()
235 _input_modulation_gate_input.allocator()->allocate(); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.cpp | 54 …rget_gate_input(), _input_gate_input(), _output_gate_input(), _input_modulation_gate_input(), _for… in NELSTMLayerQuantized()
168 _memory_group.manage(&_input_modulation_gate_input); in configure()
169 …_slice_cell_tensor.configure(&_output_lowp, &_input_modulation_gate_input, { 2 * output_size, 0 },… in configure()
180 _memory_group.manage(&_input_modulation_gate_input); in configure()
181 …_slice_cell_tensor.configure(&_output_lowp, &_input_modulation_gate_input, { 2 * output_size }, { … 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()
203 _input_modulation_gate_input.allocator()->allocate(); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.h | 201 Tensor _input_modulation_gate_input; variable
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.h | 224 CLTensor _input_modulation_gate_input; variable
|