Searched refs:_input_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 …te_input(), _input_modulation_gate_input(), _forget_gate_output(), _input_gate_output(), _output_g… in CLLSTMLayerQuantized()
226 _memory_group.manage(&_input_gate_output); in configure()
227 …_input_gate_output.allocator()->init(TensorInfo(_input_gate_input.info()->tensor_shape(), 1, DataT… in configure()
228 …_sigmoid_input_gate.configure(compile_context, &_input_gate_input, &_input_gate_output, Activation… in configure()
250 …_cell_state_tmp2.allocator()->init(TensorInfo(_input_gate_output.info()->tensor_shape(), 1, DataTy… in configure()
251 …_mul_input_gate_input_mod_gate.configure(compile_context, &_input_gate_output, &_input_modulation_… in configure()
253 _input_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()
194 _memory_group.manage(&_input_gate_output); in configure()
195 …_input_gate_output.allocator()->init(TensorInfo(_input_gate_input.info()->tensor_shape(), 1, DataT… in configure()
196 …_sigmoid_input_gate.configure(&_input_gate_input, &_input_gate_output, ActivationLayerInfo(Activat… in configure()
218 …_cell_state2.allocator()->init(TensorInfo(_input_gate_output.info()->tensor_shape(), 1, DataType::… in configure()
219 …_mul2.configure(&_input_gate_output, &_input_modulation_gate_output, &_cell_state2, 1, ConvertPoli… in configure()
221 _input_gate_output.allocator()->allocate(); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| H A D | NELSTMLayerQuantized.h | 203 Tensor _input_gate_output; variable
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| H A D | CLLSTMLayerQuantized.h | 226 CLTensor _input_gate_output; variable
|