Searched refs:_output_layer_norm_out1 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 51 …get_layer_norm_out2(), _cell_layer_norm_out1(), _cell_layer_norm_out2(), _output_layer_norm_out1(), in NELSTMLayer()
309 …_output_layer_norm_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type… in configure()
311 _memory_group.manage(&_output_layer_norm_out1); in configure()
314 …igure(output_gate_out, lstm_params.output_layer_norm_weights(), &_output_layer_norm_out1, 1, Conve… in configure()
317 …_accum_output_gate_bias.configure(&_output_layer_norm_out1, output_gate_bias, &_output_layer_norm_… in configure()
318 _output_layer_norm_out1.allocator()->allocate(); 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()
334 …_output_layer_norm_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type… in configure()
336 _memory_group.manage(&_output_layer_norm_out1); in configure()
339 …text, output_gate_out, lstm_params.output_layer_norm_weights(), &_output_layer_norm_out1, 1, Conve… in configure()
343 …_accum_output_gate_bias.configure(compile_context, &_output_layer_norm_out1, output_gate_bias, &_o… in configure()
344 _output_layer_norm_out1.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 241 Tensor _output_layer_norm_out1; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 305 CLTensor _output_layer_norm_out1; variable
|