Searched refs:_forget_layer_norm_out2 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 51 …orm_out1(), _input_layer_norm_out2(), _forget_layer_norm_out1(), _forget_layer_norm_out2(), _cell_… in NELSTMLayer()
144 …_forget_layer_norm_out2.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type… in configure()
146 _memory_group.manage(&_forget_layer_norm_out2); in configure()
151 …gate_bias.configure(&_forget_layer_norm_out1, forget_gate_bias, &_forget_layer_norm_out2, ConvertP… in configure()
153 forget_gate_out = &_forget_layer_norm_out2; in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 52 …_layer_norm_out1(), _input_layer_norm_out2(), _forget_layer_norm_out1(), _forget_layer_norm_out2(), in CLLSTMLayer()
160 …_forget_layer_norm_out2.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type… in configure()
162 _memory_group.manage(&_forget_layer_norm_out2); in configure()
168 …re(compile_context, &_forget_layer_norm_out1, forget_gate_bias, &_forget_layer_norm_out2, ConvertP… in configure()
170 forget_gate_out = &_forget_layer_norm_out2; in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 238 Tensor _forget_layer_norm_out2; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 302 CLTensor _forget_layer_norm_out2; variable
|