Searched refs:_input_gate_output (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.cpp | 64 …_input_gate_output(), _output_gate_output(), _input_modulation_gate_output(), _cell_state1(), _cel… in NELSTMLayerQuantized()
193 _memory_group.manage(&_input_gate_output); in configure()
194 …_input_gate_output.allocator()->init(TensorInfo(_input_gate_input.info()->tensor_shape(), 1, DataT… in configure()
195 …_sigmoid_input_gate.configure(&_input_gate_input, &_input_gate_output, ActivationLayerInfo(Activat… in configure()
217 …_cell_state2.allocator()->init(TensorInfo(_input_gate_output.info()->tensor_shape(), 1, DataType::… in configure()
218 …_mul2.configure(&_input_gate_output, &_input_modulation_gate_output, &_cell_state2, 1, ConvertPoli… in configure()
220 _input_gate_output.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.cpp | 60 …te_input(), _input_modulation_gate_input(), _forget_gate_output(), _input_gate_output(), _output_g… in CLLSTMLayerQuantized()
221 _memory_group.manage(&_input_gate_output); in configure()
222 …_input_gate_output.allocator()->init(TensorInfo(_input_gate_input.info()->tensor_shape(), 1, DataT… in configure()
223 …_sigmoid_input_gate.configure(compile_context, &_input_gate_input, &_input_gate_output, Activation… in configure()
245 …_cell_state_tmp2.allocator()->init(TensorInfo(_input_gate_output.info()->tensor_shape(), 1, DataTy… in configure()
246 …_mul_input_gate_input_mod_gate.configure(compile_context, &_input_gate_output, &_input_modulation_… in configure()
248 _input_gate_output.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.h | 195 Tensor _input_gate_output; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.h | 218 CLTensor _input_gate_output; variable
|