Searched refs:_cell_layer_norm_out2 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 51 …_norm_out1(), _forget_layer_norm_out2(), _cell_layer_norm_out1(), _cell_layer_norm_out2(), _output… in NELSTMLayer()
243 …_cell_layer_norm_out2.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type()… in configure()
245 _memory_group.manage(&_cell_layer_norm_out2); in configure()
250 …_accum_cell_gate_bias.configure(&_cell_layer_norm_out1, cell_bias, &_cell_layer_norm_out2, Convert… in configure()
252 cell_state_out_ptr = &_cell_layer_norm_out2; in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 53 …_cell_layer_norm_out1(), _cell_layer_norm_out2(), _output_layer_norm_out1(), _output_layer_norm_ou… in CLLSTMLayer()
265 …_cell_layer_norm_out2.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type()… in configure()
267 _memory_group.manage(&_cell_layer_norm_out2); in configure()
273 …as.configure(compile_context, &_cell_layer_norm_out1, cell_bias, &_cell_layer_norm_out2, ConvertPo… in configure()
275 cell_state_out_ptr = &_cell_layer_norm_out2; in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 240 Tensor _cell_layer_norm_out2; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 304 CLTensor _cell_layer_norm_out2; variable
|