diff options
author | 2017-07-13 02:37:33 -0700 | |
---|---|---|
committer | 2017-07-13 02:42:10 -0700 | |
commit | 896cedfa89b1693a13e5c1b32ac4ae425b3c963e (patch) | |
tree | 91f781abc1d6009b4dc7ee84e72bfc7ae14305f7 /tensorflow/core/kernels/scatter_nd_op.cc | |
parent | 79147b44135df9c2e5acbe2aaa106fba65612011 (diff) |
Automated g4 rollback of changelist 161726749
PiperOrigin-RevId: 161781962
Diffstat (limited to 'tensorflow/core/kernels/scatter_nd_op.cc')
-rw-r--r-- | tensorflow/core/kernels/scatter_nd_op.cc | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/scatter_nd_op.cc b/tensorflow/core/kernels/scatter_nd_op.cc index 59f690e7aa..1428546d52 100644 --- a/tensorflow/core/kernels/scatter_nd_op.cc +++ b/tensorflow/core/kernels/scatter_nd_op.cc @@ -26,7 +26,7 @@ limitations under the License. #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/kernels/bounds_check.h" -#include "tensorflow/core/kernels/dense_update_functor.h" +#include "tensorflow/core/kernels/dense_update_ops.h" #include "tensorflow/core/kernels/fill_functor.h" #include "tensorflow/core/lib/strings/str_util.h" #include "tensorflow/core/platform/mutex.h" @@ -523,6 +523,16 @@ TF_CALL_GPU_NUMBER_TYPES_NO_HALF(DECLARE_GPU_SPECS); #undef DECLARE_GPU_SPECS_INDEX #undef DECLARE_GPU_SPECS_INDEX_OP +#define REGISTER_GPU_KERNELS(type) \ + template <> \ + void DenseUpdate<GPUDevice, type, ASSIGN>::operator()( \ + const GPUDevice& d, typename TTypes<type>::Flat lhs, \ + typename TTypes<type>::ConstFlat rhs); \ + extern template struct DenseUpdate<GPUDevice, type, ASSIGN>; + +TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS); +#undef REGISTER_GPU_KERNELS + } // namespace functor #endif // GOOGLE_CUDA |