Searched defs:cell_state_out (Results 1 – 12 of 12) sorted by relevance
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 65 …ICLTensor *scratch_buffer, ICLTensor *output_state_out, ICLTensor *cell_state_out, ICLTensor *outp… in configure()
78 …ICLTensor *scratch_buffer, ICLTensor *output_state_out, ICLTensor *cell_state_out, ICLTensor *outp… in configure()
405 …rInfo *scratch_buffer, const ITensorInfo *output_state_out, const ITensorInfo *cell_state_out, con… in validate()
|
| D | CLLSTMLayerQuantized.cpp | 65 ICLTensor *cell_state_out, ICLTensor *output_state_out) in configure()
77 ICLTensor *cell_state_out, ICLTensor *output_state_out) in configure()
285 … const ITensorInfo *cell_state_out, const ITensorInfo *output_state_out) in validate()
|
| D | CLQLSTMLayer.cpp | 164 … ICLTensor *cell_state_out, ICLTensor *output_state_out, ICLTensor *output, in configure()
177 … ICLTensor *cell_state_out, ICLTensor *output_state_out, ICLTensor *output, in configure()
608 … const ITensorInfo *cell_state_out, const ITensorInfo *output_state_out, const ITensorInfo *output, in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 62 … ITensor *scratch_buffer, ITensor *output_state_out, ITensor *cell_state_out, ITensor *output, in configure()
380 …rInfo *scratch_buffer, const ITensorInfo *output_state_out, const ITensorInfo *cell_state_out, con… in validate()
|
| D | NELSTMLayerQuantized.cpp | 65 ITensor *cell_state_out, ITensor *output_state_out) in configure()
253 … const ITensorInfo *cell_state_out, const ITensorInfo *output_state_out) in validate()
|
| D | NEQLSTMLayer.cpp | 219 ITensor *cell_state_out, ITensor *output_state_out, ITensor *output, in configure()
680 … const ITensorInfo *cell_state_out, const ITensorInfo *output_state_out, const ITensorInfo *output, in validate()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonQuantizedLstmWorkload.cpp | 62 …arm_compute::ITensor& cell_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0])-… in NeonQuantizedLstmWorkload() local
|
| D | NeonLstmFloatWorkload.cpp | 138 …arm_compute::ITensor& cell_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[2])-… in NeonLstmFloatWorkload() local
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 256 …arm_compute::ITensor& cell_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[2])->GetTens… in NeonUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClLstmFloatWorkload.cpp | 141 …arm_compute::ICLTensor& cell_state_out = static_cast<IClTensorHandle*>(m_Data.m_Outputs[2])… in ClLstmFloatWorkload() local
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 258 …arm_compute::ICLTensor& cell_state_out = static_cast<IClTensorHandle*>(m_Data.m_Inputs[2])->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 101 TensorType cell_state_out = create_tensor<TensorType>(output_cell_shape, data_type); in compute_target() local
|