Home
last modified time | relevance | path

Searched refs:concatOutputTensorInfo (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp435 TensorInfo concatOutputTensorInfo = outputInfo; in NeonUnidirectionalSequenceLstmFloatWorkload() local
436 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmFloatWorkload()
437 BuildArmComputeTensor(concat_out, concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()
802 TensorInfo concatOutputTensorInfo = TensorInfo(output); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
803 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
804 …m_compute::TensorInfo aclConcatOutputTensorInfo= BuildArmComputeTensorInfo(concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp416 TensorInfo concatOutputTensorInfo = outputInfo; in NeonUnidirectionalSequenceLstmWorkload() local
417 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmWorkload()
418 BuildArmComputeTensor(concat_out, concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmWorkload()
775 TensorInfo concatOutputTensorInfo = TensorInfo(output); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
776 concatOutputTensorInfo.SetShape(timeMajorShapeOutput); in NeonUnidirectionalSequenceLstmWorkloadValidate()
777 …m_compute::TensorInfo aclConcatOutputTensorInfo= BuildArmComputeTensorInfo(concatOutputTensorInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()