Searched defs:outputLayerShape (Results 1 – 3 of 3) sorted by relevance
55 TensorShape outputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in NeonUnidirectionalSequenceLstmWorkload() local501 TensorShape outputLayerShape = output.GetShape(); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
52 TensorShape outputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkload() local520 TensorShape outputLayerShape = outputStateIn.GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
54 TensorShape outputLayerShape = static_cast<IClTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in ClUnidirectionalSequenceLstmFloatWorkload() local520 TensorShape outputLayerShape = outputStateIn.GetShape(); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local