Commit e6921fdc authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Add support for ResourceVariables in Shampoo optimizer. 

Updates Shampoo default matrix size to ones that work better on TPUs.
Change norm to use the F norm.

PiperOrigin-RevId: 207967042
parent 96a77055
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment