Searched refs:_cell_state_out1 (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 50 …_cell_state_out1(), _cell_state_out2(), _cell_state_out3(), _cell_state_out4(), _cell_state_out5()… in NELSTMLayer()
224 _cell_state_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
230 _memory_group.manage(&_cell_state_out1); in configure()
231 …gure(input, input_to_cell_weights, (_is_layer_norm_lstm) ? nullptr : cell_bias, &_cell_state_out1); in configure()
238 …_accum_cell_state1.configure(&_cell_state_out1, &_cell_state_out3, &_cell_state_out4, ConvertPolic… in configure()
259 …_accum_cell_state2.configure(&_cell_state_out5, &_cell_state_out3, &_cell_state_out1, ConvertPolic… in configure()
266 …_cell_clip.configure(&_cell_state_out1, nullptr, ActivationLayerInfo(ActivationLayerInfo::Activati… in configure()
292 …_output3.allocator()->init(TensorInfo(_cell_state_out1.info()->tensor_shape(), 1, input->info()->d… in configure()
295 …_pixelwise_mul_output_state1.configure(&_cell_state_out1, lstm_params.cell_to_output_weights(), &_… in configure()
337 _activation_output_state.configure(&_cell_state_out1, &_cell_state_activation, activation_info); in configure()
[all …]
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 51 …3(), _forget_gate_out4(), _forget_gate_out5(), _forget_gate_out6(), _cell_state_out1(), _cell_stat… in CLLSTMLayer()
245 _cell_state_out1.allocator()->init(TensorInfo(cell_state_shape, 1, input->info()->data_type())); in configure()
251 _memory_group.manage(&_cell_state_out1); in configure()
252 …ext, input, input_to_cell_weights, (_is_layer_norm_lstm) ? nullptr : cell_bias, &_cell_state_out1); in configure()
260 …_accum_cell_state1.configure(compile_context, &_cell_state_out1, &_cell_state_out3, &_cell_state_o… in configure()
282 …2.configure(compile_context, &_cell_state_out5, &_cell_state_out3, &_cell_state_out1, ConvertPolic… in configure()
289 …_cell_clip.configure(compile_context, &_cell_state_out1, nullptr, ActivationLayerInfo(ActivationLa… in configure()
317 …_output3.allocator()->init(TensorInfo(_cell_state_out1.info()->tensor_shape(), 1, input->info()->d… in configure()
320 …_pixelwise_mul_output_state1.configure(compile_context, &_cell_state_out1, lstm_params.cell_to_out… in configure()
363 …_activation_output_state.configure(compile_context, &_cell_state_out1, &_cell_state_activation, ac… in configure()
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 223 Tensor _cell_state_out1; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 287 CLTensor _cell_state_out1; variable
|