experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[core] scatter_op: Remove duplicate check for IsScalar()

Browse Source
This commit is contained in:
Gaurav Singh 2020-01-29 20:45:15 -05:00
parent 2f4812db43
commit 4edec1af39
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/kernels/scatter_op.cc
View File

@ -124,8 +124,7 @@ class ScatterUpdateOp : public OpKernel {
auto indices_flat = indices.flat<Index>();
auto params_flat = params.flat_outer_dims<T>();
if (TensorShapeUtils::IsScalar(updates.shape()) ||
TensorShapeUtils::IsScalar(updates.shape())) {
if (TensorShapeUtils::IsScalar(updates.shape())) {
const auto update = updates.scalar<T>();
functor::ScatterScalarFunctor<Device, T, Index, op> functor;
const Index bad_i = functor(c, c->template eigen_device<Device>(),