Home
last modified time | relevance | path

Searched defs:output_state_out (Results 1 – 12 of 12) sorted by relevance

/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayer.cpp65 …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()
DCLLSTMLayerQuantized.cpp65 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()
DCLQLSTMLayer.cpp164 … 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/
DNELSTMLayer.cpp62 … 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()
DNELSTMLayerQuantized.cpp65 ITensor *cell_state_out, ITensor *output_state_out) in configure()
253 … const ITensorInfo *cell_state_out, const ITensorInfo *output_state_out) in validate()
DNEQLSTMLayer.cpp219 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/
DNeonQuantizedLstmWorkload.cpp63 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])-… in NeonQuantizedLstmWorkload() local
DNeonLstmFloatWorkload.cpp137 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])-… in NeonLstmFloatWorkload() local
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp255 …arm_compute::ITensor& output_state_out = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTe… in NeonUnidirectionalSequenceLstmFloatWorkload() local
/external/armnn/src/backends/cl/workloads/
DClLstmFloatWorkload.cpp140 …arm_compute::ICLTensor& output_state_out = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])… in ClLstmFloatWorkload() local
DClUnidirectionalSequenceLstmFloatWorkload.cpp257 …arm_compute::ICLTensor& output_state_out = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetT… in ClUnidirectionalSequenceLstmFloatWorkload() local
/external/ComputeLibrary/tests/validation/fixtures/
DLSTMLayerFixture.h100 TensorType output_state_out = create_tensor<TensorType>(output_shape, data_type); in compute_target() local