experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #30239 from benbarsdell:amp-add-batchmatmulv2

Browse Source 
PiperOrigin-RevId: 255699129
This commit is contained in:
TensorFlower Gardener 2019-06-28 21:35:15 -07:00
commit dcbf7f7bc0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
View File

@ -62,7 +62,7 @@ class AutoMixedPrecisionLists {
auto list = gtl::FlatSet<string> {
#if CUDA_VERSION >= 9010 // Fp16 BatchMatMul is slow before CUDA 9.1.
"BatchMatMul",
"BatchMatMul", "BatchMatMulV2",
#endif
"BlockLSTM", "BlockLSTMGrad", "Conv2D", "Conv2DBackpropFilter",
"Conv2DBackpropInput",