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

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

@ -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);