You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Softmax will likely only support float point types: #283
Currently, the spec provides no way to convert between data types, this might make it hard to use softmax with certain computation graphs.
This could impact models doing integer computations (e.g. image recognition models that takes in RGB24 image represented by uint8), and returns softmax-ed predictions (e.g. one-hot encoding).
This came up during review for softmax implementation https://chromium-review.googlesource.com/c/chromium/src/+/3856752/comments/267c926c_b14af87f
Softmax will likely only support float point types: #283
Currently, the spec provides no way to convert between data types, this might make it hard to use softmax with certain computation graphs.
This could impact models doing integer computations (e.g. image recognition models that takes in RGB24 image represented by uint8), and returns softmax-ed predictions (e.g. one-hot encoding).
For example, the following graph:
Perhaps WebNN needs a type-casting function?
In ML frameworks:
The text was updated successfully, but these errors were encountered: