Searched refs:permuteOutInfo (Results 1 – 3 of 3) sorted by relevance
70 TensorInfo permuteOutInfo = inputInfo; in NeonUnidirectionalSequenceLstmFloatWorkload() local71 permuteOutInfo.SetShape(timeMajorShapeInput); in NeonUnidirectionalSequenceLstmFloatWorkload()72 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()547 TensorInfo permuteOutInfo = TensorInfo(input); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local548 …e::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permuteOutInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
72 TensorInfo permuteOutInfo = inputInfo; in NeonUnidirectionalSequenceLstmWorkload() local73 permuteOutInfo.SetShape(timeMajorShapeInput); in NeonUnidirectionalSequenceLstmWorkload()74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmWorkload()528 TensorInfo permuteOutInfo = TensorInfo(input); in NeonUnidirectionalSequenceLstmWorkloadValidate() local529 …e::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permuteOutInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()
72 TensorInfo permuteOutInfo = inputInfo; in ClUnidirectionalSequenceLstmFloatWorkload() local73 permuteOutInfo.SetShape(timeMajorShapeInput); in ClUnidirectionalSequenceLstmFloatWorkload()74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in ClUnidirectionalSequenceLstmFloatWorkload()547 TensorInfo permuteOutInfo = TensorInfo(input); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local548 …e::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permuteOutInfo); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()