Searched defs:output_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 …const ITensorInfo *scratch_buffer, const ITensorInfo *output_state_out, const ITensorInfo *cell_st… 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 …const ITensorInfo *scratch_buffer, const ITensorInfo *output_state_out, const ITensorInfo *cell_st… 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 | 63 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])-… in NeonQuantizedLstmWorkload() local
|
| D | NeonLstmFloatWorkload.cpp | 137 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])-… in NeonLstmFloatWorkload() local
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 255 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTe… in NeonUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClLstmFloatWorkload.cpp | 140 …arm_compute::ICLTensor& output_state_out = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])… in ClLstmFloatWorkload() local
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 257 …arm_compute::ICLTensor& output_state_out = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetT… in ClUnidirectionalSequenceLstmFloatWorkload() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 100 TensorType output_state_out = create_tensor<TensorType>(output_shape, data_type); in compute_target() local
|