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
|