diff options
Diffstat (limited to 'tensorflow/core/ops/math_grad.cc')
-rw-r--r-- | tensorflow/core/ops/math_grad.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/ops/math_grad.cc b/tensorflow/core/ops/math_grad.cc index 15bcd53322..a530d286f7 100644 --- a/tensorflow/core/ops/math_grad.cc +++ b/tensorflow/core/ops/math_grad.cc @@ -588,6 +588,7 @@ REGISTER_OP_GRADIENT("Mean", MeanGrad); // REGISTER_OP_GRADIENT("SegmentMin", SegmentMinGrad); // REGISTER_OP_GRADIENT("SegmentMax", SegmentMaxGrad); // REGISTER_OP_GRADIENT("UnsortedSegmentSum", UnsortedSegmentSumGrad); +// REGISTER_OP_GRADIENT("UnsortedSegmentMax", UnsortedSegmentMaxGrad); Status MinMaxGradHelper(const string& op, const AttrSlice& attrs, FunctionDef* g) { |