From 4edec1af39565bd512bd2dd31c79e96766bf60a0 Mon Sep 17 00:00:00 2001 From: Gaurav Singh Date: Wed, 29 Jan 2020 20:45:15 -0500 Subject: [PATCH] [core] scatter_op: Remove duplicate check for IsScalar() --- tensorflow/core/kernels/scatter_op.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/core/kernels/scatter_op.cc b/tensorflow/core/kernels/scatter_op.cc index 6eae1b7e217..1c1f0d7e9e0 100644 --- a/tensorflow/core/kernels/scatter_op.cc +++ b/tensorflow/core/kernels/scatter_op.cc @@ -124,8 +124,7 @@ class ScatterUpdateOp : public OpKernel { auto indices_flat = indices.flat(); auto params_flat = params.flat_outer_dims(); - if (TensorShapeUtils::IsScalar(updates.shape()) || - TensorShapeUtils::IsScalar(updates.shape())) { + if (TensorShapeUtils::IsScalar(updates.shape())) { const auto update = updates.scalar(); functor::ScatterScalarFunctor functor; const Index bad_i = functor(c, c->template eigen_device(),