Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayer.cpp64 const ICLTensor *output_state_in, ICLTensor *cell_state_in, in configure()
77 const ICLTensor *output_state_in, ICLTensor *cell_state_in, in configure()
404 const ITensorInfo *output_state_in, const ITensorInfo *cell_state_in, in validate()
DCLLSTMLayerQuantized.cpp64 ICLTensor *cell_state_in, const ICLTensor *output_state_in, in configure()
76 ICLTensor *cell_state_in, const ICLTensor *output_state_in, in configure()
284 … const ITensorInfo *cell_state_in, const ITensorInfo *output_state_in, in validate()
DCLQLSTMLayer.cpp163 ICLTensor *cell_state_in, ICLTensor *output_state_in, in configure()
176 ICLTensor *cell_state_in, ICLTensor *output_state_in, in configure()
607 const ITensorInfo *cell_state_in, const ITensorInfo *output_state_in, in validate()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayer.cpp61 const ITensor *output_state_in, const ITensor *cell_state_in, in configure()
379 const ITensorInfo *output_state_in, const ITensorInfo *cell_state_in, in validate()
DNELSTMLayerQuantized.cpp64 ITensor *cell_state_in, const ITensor *output_state_in, in configure()
252 … const ITensorInfo *cell_state_in, const ITensorInfo *output_state_in, in validate()
DNEQLSTMLayer.cpp218 const ITensor *cell_state_in, ITensor *output_state_in, in configure()
679 const ITensorInfo *cell_state_in, const ITensorInfo *output_state_in, in validate()
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp60 …const arm_compute::ITensor& output_state_in = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[2])->… in NeonQuantizedLstmWorkload() local
DNeonLstmFloatWorkload.cpp134 …const arm_compute::ITensor& output_state_in = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[1])->… in NeonLstmFloatWorkload() local
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp252 …arm_compute::ITensor& output_state_in = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload() local
/external/armnn/src/backends/cl/workloads/
DClLstmFloatWorkload.cpp137 …const arm_compute::ICLTensor& output_state_in = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])-… in ClLstmFloatWorkload() local
DClUnidirectionalSequenceLstmFloatWorkload.cpp254 …arm_compute::ICLTensor& output_state_in = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTe… in ClUnidirectionalSequenceLstmFloatWorkload() local
/external/ComputeLibrary/tests/validation/fixtures/
DLSTMLayerFixture.h97 TensorType output_state_in = create_tensor<TensorType>(output_shape, data_type); in compute_target() local