Commit 832ffc71 authored by Yangzihao Wang's avatar Yangzihao Wang Committed by TensorFlower Gardener
Browse files

Add new env-var TF_FP16_CONV_MODE.

The env-var is used to decide how to do internal computation for cudnn convolution when input data type is float16.
For ACCURATE mode, we always use float32 as the internal compute type;
For FAST mode, we include both float16 and float32 internal compute type into auto-tune to pick whichever runs faster.

PiperOrigin-RevId: 174495814
parent 509d0f2c
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment