Commit 2451eef1 authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Fix bug where custom layers could crash. 

Layer.add_weight would crash when called without a dtype or initializer.

PiperOrigin-RevId: 196583182
parent 1761e1dd
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