flax
add promote_dtype to all standard layers
#5080
Merged

add promote_dtype to all standard layers #5080

cgarciae
cgarciae cgarciae force pushed from 9520effe to 8f867fcb 145 days ago
IvyZX
IvyZX approved these changes on 2025-11-10
cgarciae cgarciae added pull ready
cgarciae cgarciae force pushed from 8f867fcb to 74e75533 142 days ago
cgarciae cgarciae force pushed from 74e75533 to 927e1dca 142 days ago
cgarciae add promote_dtype to all standard layers
3a630ff0
cgarciae cgarciae force pushed from 927e1dca to 3a630ff0 142 days ago
copybara-service copybara-service merged 49ef97c3 into main 141 days ago
copybara-service copybara-service deleted the improve-promote-dtype-support branch 141 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone