Commit 8d327187 authored by Ben Barsdell's avatar Ben Barsdell Committed by Appledore
Browse files

Enable CUDNN_TENSOR_OP_MATH for fp16 RNNs (#17367) 

- Speeds up CUDNN RNNs with fp16 input/output when possible on supported
  GPUs. Computations will fall back to pseudo-fp16 if tensor op math is
  not supported.
- Enabled by default, but can be disabled by setting the environment
  variable TF_DISABLE_CUDNN_RNN_TENSOR_OP_MATH=1.
parent ae3badb0
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