Searched refs:_input_gate_input (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.cpp | 54 …(), _output_highp(), _output_lowp(), _bias(), _forget_gate_input(), _input_gate_input(), _output_g… in NELSTMLayerQuantized()
164 _memory_group.manage(&_input_gate_input); in configure()
165 …_slice_input_tensor.configure(&_output_lowp, &_input_gate_input, { 0, 0 }, { output_size, batch_si… in configure()
176 _memory_group.manage(&_input_gate_input); in configure()
177 _slice_input_tensor.configure(&_output_lowp, &_input_gate_input, { 0 }, { output_size }); 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()
197 _input_gate_input.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.cpp | 54 …_transposed(), _output_highp(), _output_lowp(), _bias(), _forget_gate_input(), _input_gate_input(), in CLLSTMLayerQuantized()
196 _memory_group.manage(&_input_gate_input); in configure()
197 …_slice_input_tensor.configure(compile_context, &_output_lowp, &_input_gate_input, { 0, 0 }, { outp… in configure()
208 _memory_group.manage(&_input_gate_input); in configure()
209 …_slice_input_tensor.configure(compile_context, &_output_lowp, &_input_gate_input, { 0 }, { 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()
229 _input_gate_input.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.h | 199 Tensor _input_gate_input; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.h | 222 CLTensor _input_gate_input; variable
|