Searched refs:_forget_gate_out1 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 217 Tensor _forget_gate_out1; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 281 CLTensor _forget_gate_out1; variable
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 49 …_input_gate_out2(), _input_gate_out3(), _input_gate_out4(), _forget_gate_out1(), _forget_gate_out2… in NELSTMLayer()
100 … _forget_gate_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
120 _memory_group.manage(&_forget_gate_out1); in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 50 …out1(), _input_gate_out2(), _input_gate_out3(), _input_gate_out4(), _forget_gate_out1(), _forget_g… in CLLSTMLayer()
112 … _forget_gate_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
136 _memory_group.manage(&_forget_gate_out1); in configure()
|