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

Fix support for seq2seq with mixed precision 

When the type of the input tensor `x` is not the same as the type of
the hidden states cast is required.

This mixed precision case occurs when using the seq2seq
layer with a data type of float16 or bfloat16.

PiperOrigin-RevId: 204364209
parent c61c7f1e
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