diff options
Diffstat (limited to 'tensorflow/python/eager/pywrap_tfe_src.cc')
| -rw-r--r-- | tensorflow/python/eager/pywrap_tfe_src.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/eager/pywrap_tfe_src.cc b/tensorflow/python/eager/pywrap_tfe_src.cc index 6d3ef9a37b..9789dbadee 100644 --- a/tensorflow/python/eager/pywrap_tfe_src.cc +++ b/tensorflow/python/eager/pywrap_tfe_src.cc @@ -1836,6 +1836,8 @@ bool OpGradientDoesntRequireOutputIndices( {"SoftplusGrad", {true, {}}}, {"Softsign", {true, {}}}, {"ReluGrad", {true, {}}}, + {"LeakyRelu", {true, {}}}, + {"LeakyReluGrad", {true, {}}}, {"Conv2D", {true, {}}}, {"DepthwiseConv2dNative", {true, {}}}, {"Dilation2D", {true, {}}}, |