PR #8509 Added softmax to autocast policy

Added softmax to autocast policy #8509

lsy323 merged 3 commits into master from autocast_sorftmax

added softmax to autocast policy

ad174034

fixed formatting

67882907

avizon-aws requested a review from

cowanmeg 1 year ago

lsy323 added tpuci

force push

da71b184

lsy323 approved these changes on 2024-12-20

lsy323 merged 38d08689 into master 1 year ago

Reviewers

lsy323

cowanmeg

Assignees

No one assigned

Labels

None yet

Milestone

No milestone