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

Add newer version of ops to gradient exclusion list

Browse Source 
PiperOrigin-RevId: 290645847
Change-Id: I6057a7bc93ade001b976aa97186263b9cfc6368e
This commit is contained in:
Gaurav Jain 2020-01-20 13:46:23 -08:00 committed by TensorFlower Gardener
parent 732986d06b
commit 20b2863cee
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/eager/pywrap_tfe_src.cc
View File

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