Merge pull request #37914 from benbarsdell:amp-add-einsum

PiperOrigin-RevId: 303038314 Change-Id: I25b442f965d8b927a1e3680a2b9078075e3bb849
2020-03-25 21:54:16 -07:00 · 2020-03-25 21:54:16 -07:00 · 268f9834b6
commit 268f9834b6
parent 028505ae19 0107439506
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
+++ b/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
@ -69,6 +69,7 @@ class AutoMixedPrecisionLists {
        "CudnnRNNBackpropV3",
        "CudnnRNNV2",
        "CudnnRNNV3",
+        "Einsum",
        "GRUBlockCell",
        "GRUBlockCellGrad",
        "LSTMBlockCell",