From 20b2863cee014625dcf81498a6deac3873f81879 Mon Sep 17 00:00:00 2001
From: Gaurav Jain <gjn@google.com>
Date: Mon, 20 Jan 2020 13:46:23 -0800
Subject: [PATCH] Add newer version of ops to gradient exclusion list

PiperOrigin-RevId: 290645847
Change-Id: I6057a7bc93ade001b976aa97186263b9cfc6368e
---
 tensorflow/python/eager/pywrap_tfe_src.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tensorflow/python/eager/pywrap_tfe_src.cc b/tensorflow/python/eager/pywrap_tfe_src.cc
index 3c4e8d72a33..6e8762f8315 100644
--- a/tensorflow/python/eager/pywrap_tfe_src.cc
+++ b/tensorflow/python/eager/pywrap_tfe_src.cc
@@ -2957,6 +2957,7 @@ bool OpGradientDoesntRequireOutputIndices(
           {"Cos", {true, {}}},
           {"Tan", {true, {}}},
           {"Add", {true, {}}},
+          {"AddV2", {true, {}}},
           {"Sub", {true, {}}},
           {"Mul", {true, {}}},
           {"Div", {true, {}}},
@@ -2984,6 +2985,8 @@ bool OpGradientDoesntRequireOutputIndices(
 
           // Ops that don't require a subset of outputs.
           {"FusedBatchNorm", {false, {0, 1, 2}}},
+          {"FusedBatchNormV2", {false, {0, 1, 2}}},
+          {"FusedBatchNormV3", {false, {0, 1, 2}}},
       });
 
   auto it = m->find(op_name);
@@ -3031,6 +3034,8 @@ bool OpGradientDoesntRequireInputIndices(
 
           // Ops that don't require a subset of inputs.
           {"FusedBatchNorm", {false, {2}}},
+          {"FusedBatchNormV2", {false, {2}}},
+          {"FusedBatchNormV3", {false, {2}}},
       });
 
   auto it = m->find(op_name);