Searched refs:inputLayerShape (Results 1 – 3 of 3) sorted by relevance
50 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkload() local54 … unsigned int maxTime = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[0] : inputLayerShape[1]; in NeonUnidirectionalSequenceLstmFloatWorkload()55 …unsigned int batchSize = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[1] : inputLayerShape[0]; in NeonUnidirectionalSequenceLstmFloatWorkload()56 unsigned int inputSize = inputLayerShape[2]; in NeonUnidirectionalSequenceLstmFloatWorkload()519 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local522 unsigned int maxTime = descriptor.m_TimeMajor ? inputLayerShape[0] : inputLayerShape[1]; in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()523 unsigned int batchSize = descriptor.m_TimeMajor ? inputLayerShape[1] : inputLayerShape[0]; in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()524 unsigned int inputSize = inputLayerShape[2]; in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
54 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmWorkload() local57 … unsigned int maxTime = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[0] : inputLayerShape[1]; in NeonUnidirectionalSequenceLstmWorkload()58 …unsigned int batchSize = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[1] : inputLayerShape[0]; in NeonUnidirectionalSequenceLstmWorkload()59 unsigned int inputSize = inputLayerShape[2]; in NeonUnidirectionalSequenceLstmWorkload()500 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmWorkloadValidate() local503 unsigned int maxTime = descriptor.m_TimeMajor ? inputLayerShape[0] : inputLayerShape[1]; in NeonUnidirectionalSequenceLstmWorkloadValidate()504 unsigned int batchSize = descriptor.m_TimeMajor ? inputLayerShape[1] : inputLayerShape[0]; in NeonUnidirectionalSequenceLstmWorkloadValidate()505 unsigned int inputSize = inputLayerShape[2]; in NeonUnidirectionalSequenceLstmWorkloadValidate()
52 TensorShape inputLayerShape = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in ClUnidirectionalSequenceLstmFloatWorkload() local56 … unsigned int maxTime = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[0] : inputLayerShape[1]; in ClUnidirectionalSequenceLstmFloatWorkload()57 …unsigned int batchSize = m_Data.m_Parameters.m_TimeMajor ? inputLayerShape[1] : inputLayerShape[0]; in ClUnidirectionalSequenceLstmFloatWorkload()58 unsigned int inputSize = inputLayerShape[2]; in ClUnidirectionalSequenceLstmFloatWorkload()519 TensorShape inputLayerShape = input.GetShape(); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local522 unsigned int maxTime = descriptor.m_TimeMajor?inputLayerShape[0]:inputLayerShape[1]; in ClUnidirectionalSequenceLstmFloatWorkloadValidate()523 unsigned int batchSize = descriptor.m_TimeMajor?inputLayerShape[1]:inputLayerShape[0]; in ClUnidirectionalSequenceLstmFloatWorkloadValidate()524 unsigned int inputSize = inputLayerShape[2]; in ClUnidirectionalSequenceLstmFloatWorkloadValidate()