Home
last modified time | relevance | path

Searched defs:inputLayerShape (Results 1 – 3 of 3) sorted by relevance

/external/armnn/src/backends/neon/workloads/
DNeonUnidirectionalSequenceLstmWorkload.cpp54 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmWorkload() local
500 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp50 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkload() local
519 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
/external/armnn/src/backends/cl/workloads/
DClUnidirectionalSequenceLstmFloatWorkload.cpp52 TensorShape inputLayerShape = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in ClUnidirectionalSequenceLstmFloatWorkload() local
519 TensorShape inputLayerShape = input.GetShape(); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local