Searched refs:concatOutputTensorInfo (Results 1 – 2 of 2) sorted by relevance
435 TensorInfo concatOutputTensorInfo = outputInfo; in NeonUnidirectionalSequenceLstmFloatWorkload() local436 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmFloatWorkload()437 BuildArmComputeTensor(concat_out, concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()802 TensorInfo concatOutputTensorInfo = TensorInfo(output); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local803 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()804 …m_compute::TensorInfo aclConcatOutputTensorInfo= BuildArmComputeTensorInfo(concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
416 TensorInfo concatOutputTensorInfo = outputInfo; in NeonUnidirectionalSequenceLstmWorkload() local417 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmWorkload()418 BuildArmComputeTensor(concat_out, concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmWorkload()775 TensorInfo concatOutputTensorInfo = TensorInfo(output); in NeonUnidirectionalSequenceLstmWorkloadValidate() local776 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmWorkloadValidate()777 …m_compute::TensorInfo aclConcatOutputTensorInfo= BuildArmComputeTensorInfo(concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()