Bug: Fix overflow with Softmax - Logsoftmax

[raphaelDkhn]

Pull Request type

Please check the type of change your PR introduces:

• Bugfix
• Feature
• Code style update (formatting, renaming)
• Refactoring (no functional changes, no API changes)
• Build-related changes
• Documentation content changes
• Other (please describe):

What is the current behavior?

With some neural networks, exp in (log)softmax operator create u32 overflow.

What is the new behavior?

for FP8x23 and FP16x16 tensors (log)softmax uses wide operations. It means that exp output type is upcasted, and div downcasted back.

Does this introduce a breaking change?

• Yes
• No

Other information