Searched refs:_output4 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 50 …l_state_out4(), _cell_state_out5(), _output1(), _output2(), _output3(), _output4(), _cell_state_ac… in NELSTMLayer()
274 _output4.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
282 _memory_group.manage(&_output4); in configure()
284 …gure(&_forget_gate_out2, &_output2, (_is_layer_norm_lstm) ? nullptr : output_gate_bias, &_output4); in configure()
289 Tensor *output_gate_out = &_output4; in configure()
296 _accum_output1.configure(&_output4, &_output3, &_output1, ConvertPolicy::SATURATE); in configure()
297 _output4.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 52 …_output2(), _output3(), _output4(), _cell_state_activation(), _output_state1(), _ones(), _input_la… in CLLSTMLayer()
297 _output4.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
307 _memory_group.manage(&_output4); in configure()
309 …ext, &_forget_gate_out2, &_output2, (_is_layer_norm_lstm) ? nullptr : output_gate_bias, &_output4); in configure()
314 CLTensor *output_gate_out = &_output4; in configure()
321 …_accum_output1.configure(compile_context, &_output4, &_output3, &_output1, ConvertPolicy::SATURATE… in configure()
322 _output4.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 231 Tensor _output4; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 295 CLTensor _output4; variable
|