diff options
author | 2017-12-04 00:57:14 -0800 | |
---|---|---|
committer | 2017-12-04 01:00:36 -0800 | |
commit | 88e63eb7099e2cd82e942f4b3867a9fedffb3d85 (patch) | |
tree | 7dc802f1e1af948f15639465aca4212130dea7e1 /tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc | |
parent | ed2e8c227e77d591b868ecd6d1c8d59ddae3b3d9 (diff) |
Make gather_nd & scatter_nd handle rank 6 and 7 indices.
PiperOrigin-RevId: 177781215
Diffstat (limited to 'tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc')
-rw-r--r-- | tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc b/tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc new file mode 100644 index 0000000000..9222cb0769 --- /dev/null +++ b/tensorflow/core/kernels/gather_nd_op_cpu_impl_7.cc @@ -0,0 +1,18 @@ +/* Copyright 2016 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. +==============================================================================*/ + +#define CPU_PROVIDED_IXDIM 7 +#include "tensorflow/core/kernels/gather_nd_op_cpu_impl.h" +#undef CPU_PROVIDED_IXDIM |