[MLIR][TOSA] Fix converting tosa.clamp and tosa.relu to linalg
Tosa to Linalg conversion crashes when input tensor is a float type other than fp32. Because tosa.clamp and tosa.reluN have fp32 min/max attribute which is converted as arith.constant with the attribute type. This commit fixes the crash by correctly setting the float constant type from the input tensor. Reviewed By: eric-k256 Differential Revision: https://reviews.llvm.org/D128630
Loading
Please sign in to comment