diff options
author | 2017-01-23 17:08:42 -0800 | |
---|---|---|
committer | 2017-01-23 17:30:28 -0800 | |
commit | 9b7c47c1d48dfbe69e2ab62aae6146823ba7e664 (patch) | |
tree | eb0e116588ac6b18d866d817979f32fd8eff7e60 /tensorflow/core/kernels/inplace_ops.cc | |
parent | 5aed36230aaa8e727990f1b8e37c82bed8b5356f (diff) |
Enable inplace update ops for strings on CPU. Speed up string concat.
Change: 145356129
Diffstat (limited to 'tensorflow/core/kernels/inplace_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/inplace_ops.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/inplace_ops.cc b/tensorflow/core/kernels/inplace_ops.cc index b44f2f5465..a4f2e1053d 100644 --- a/tensorflow/core/kernels/inplace_ops.cc +++ b/tensorflow/core/kernels/inplace_ops.cc @@ -48,6 +48,7 @@ Status DoParallelConcat(const CPUDevice& d, const Tensor& value, int32 loc, case DataTypeToEnum<type>::value: \ return DoParallelConcatUpdate<type>(d, value, loc, output); TF_CALL_NUMBER_TYPES(CASE); + TF_CALL_string(CASE); #undef CASE default: return errors::InvalidArgument("Unsupported data type: ", value.dtype()); @@ -132,7 +133,7 @@ class FailureKernel : public OpKernel { .Device(DEVICE_CPU) \ .TypeConstraint<type>("T"), \ ParallelConcatUpdate<CPUDevice>); -TF_CALL_NUMBER_TYPES(REGISTER) +TF_CALL_POD_STRING_TYPES(REGISTER) #undef REGISTER #define REGISTER_EMPTY(type) \ |