diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-10-28 17:45:51 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-28 19:05:47 -0700 |
commit | 3bb585a7b3592763ef5c6b11a897c1a8ff99ea81 (patch) | |
tree | cb42e1ad1b959a707aac85935a80e5abfe656c51 /tensorflow/core/kernels/scatter_nd_op.h | |
parent | 64081c872e7c617e4378135c634ffd1a24162103 (diff) |
Automated rollback of change 137564676
Change: 137576487
Diffstat (limited to 'tensorflow/core/kernels/scatter_nd_op.h')
-rw-r--r-- | tensorflow/core/kernels/scatter_nd_op.h | 52 |
1 files changed, 0 insertions, 52 deletions
diff --git a/tensorflow/core/kernels/scatter_nd_op.h b/tensorflow/core/kernels/scatter_nd_op.h deleted file mode 100644 index e4c8e7ed9f..0000000000 --- a/tensorflow/core/kernels/scatter_nd_op.h +++ /dev/null @@ -1,52 +0,0 @@ -/* Copyright 2015 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -==============================================================================*/ - -#ifndef TENSORFLOW_KERNELS_SCATTER_ND_OP_H_ -#define TENSORFLOW_KERNELS_SCATTER_ND_OP_H_ - -// Functor definitions for ScatterND ops, must be compilable by nvcc. - -#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" -#include "tensorflow/core/framework/tensor_types.h" - -namespace tensorflow { - -class OpKernelContext; - -namespace scatter_nd_op { - -enum class UpdateOp { ASSIGN, ADD, SUB, MUL, DIV }; - -} // namespace scatter_nd_op - -namespace functor { - -// Functor used by ScatterOp to do the computations. -template <typename Device, typename T, typename Index, - scatter_nd_op::UpdateOp op, int IXDIM> -struct ScatterNdFunctor { - // Returns -1 on success or a nonnegative i s.t. indices[i] is a bad index. - Index operator()(OpKernelContext* c, const Device& d, - typename TTypes<Index>::Scalar Tscratch, - typename TTypes<T, IXDIM>::Tensor params, - typename TTypes<T, 2>::ConstTensor indices, - typename TTypes<T, 2>::ConstTensor updates, - typename TTypes<T, IXDIM + 1>::Tensor Toutput); -}; - -} // namespace functor -} // namespace tensorflow - -#endif // TENSORFLOW_KERNELS_SCATTER_ND_OP_H_ |