[tosa] Change axis to I32Attr
Hi @Tai78641 Can you please resolve conflicts? Thank you!
Fix issues due to TOSA axis attribute type changing from I64Attr to I32Attr to match the TOSA spec.

this needs to be coordinated with https://reviews.llvm.org/D157424 which changes tosa dialect axis attribute to I32Attr

Change-Id: If8bfc4c4af30712226b36f79dcd600e47d0f168d