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

Automatic cast layer inputs to the layer's dtype. 

This makes it more convenient to use layer of different dtypes in a model. Instead of having to manually cast intermediate tensors between layers of different dtypes, they will automatically be casted.

This is also useful for the upcoming mixed precision API.

PiperOrigin-RevId: 200783477
parent 5e9a39d6
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