diff options
| author |  Pan Daoxin <daoxin.pan@intel.com> | 2018-09-21 09:24:42 +0800 |
|---|---|---|
| committer | Pan Daoxin <daoxin.pan@intel.com> | 2018-09-21 09:24:42 +0800 |
| commit | a3321411e4f7b5ba3f1773de8a557947a4851671 (patch) | |
| tree | c105462de626f7a58b6ef6d6d702cb86a4295618 /tensorflow/core/ops | |
| parent | d7a8e852941e8cd856caafddf7c93d857e83e8b1 (diff) | |
More changes.
Diffstat (limited to 'tensorflow/core/ops')
| -rw-r--r-- | tensorflow/core/ops/array_ops.cc | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/tensorflow/core/ops/array_ops.cc b/tensorflow/core/ops/array_ops.cc index 325690eded..15cafaddf4 100644 --- a/tensorflow/core/ops/array_ops.cc +++ b/tensorflow/core/ops/array_ops.cc @@ -1539,9 +1539,7 @@ REGISTER_OP("Slice") .Output("output: T") .Attr("T: type") .Attr("Index: {int32,int64}") - .SetShapeFn([](InferenceContext* c) { - return shape_inference::SliceShape(c); - }); + .SetShapeFn(shape_inference::SliceShape); #ifdef INTEL_MKL REGISTER_OP("_MklSlice") @@ -1555,9 +1553,7 @@ REGISTER_OP("_MklSlice") .Output("mkl_output: uint8") .Attr("T: type") .Attr("Index: {int32,int64}") - .SetShapeFn([](InferenceContext* c) { - return shape_inference::SliceShape(c); - }); + .SetShapeFn(shape_inference::SliceShape); #endif REGISTER_OP("StridedSlice") |