diff options
| author |  Nikolaas Steenbergen <nikolaas.steenbergen@googlemail.com> | 2017-02-04 00:15:21 +0100 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2017-02-03 15:15:21 -0800 |
| commit | 94f2229c9e4b4324a324330c8f419276eda7e503 (patch) | |
| tree | d458b69c476dfd42a214032f2d1e277fa4cd0d18 /tensorflow/core/ops/math_grad.cc | |
| parent | 084b37a00f3cf2cc89d433528ca63ec1d3b5b313 (diff) | |
TF-549 Adds unsorted segment max Op (#6975)
* TF-549 Adds unsorted segment max Op
* Cosmetic change
* Add todo comment
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) { |