Add MLFloat16 support for LayerNormalization, SkipLayerNormalization #22063
Add MLFloat16 support for LayerNormalization
6c53ff2c
register LayerNormalization
af042602
inline convert functions
1a6c1d84
a few renames
38a24b86
enable_if_t, is_same_v
079f0c9b
Add MLFloat16 support for SkipLayerNormalization, SkipSimplifiedLayer…
ba1bbdd4
register LayerNormalization, SimplifiedLayerNormalization in cpu_cont…
9b0de4b8
add constexpr
d2e0b912
reorder
246332ea
lint
32e8c768
Fix null check
fdc0ac46
save a cast
16f7c012
Update documentation
de49a893
amarin16
marked this pull request as ready for review 1 year ago
Merge branch 'main' into dev/amarin16/layer_norm
484dc17f
use fp32 output buffer to avoid a conversion
fc625ea6
fix check warnings
25bfdcee
fix lint error
ea6e388e
lint
134a6092
use size_t instead of int64_t
528e1e27
add cast to fix pipeline errors
a7d056c5
yufenglee
approved these changes
on 2024-09-23
amarin16
merged
eb2506d7
into main 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub