diff options
author | Eugene Brevdo <ebrevdo@google.com> | 2018-09-25 11:54:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 12:01:18 -0700 |
commit | df930015230c1195065e2fd01c61f527b8662efb (patch) | |
tree | a197a87f941777ac1a0596eeb0522c001e7e0fd3 /tensorflow/core | |
parent | 3f2d0ad596d5f2cc30f10148ac56ac5680603f5b (diff) |
[TF TensorArray] TensorSetZero supports bool types.
PiperOrigin-RevId: 214478085
Diffstat (limited to 'tensorflow/core')
-rw-r--r-- | tensorflow/core/kernels/tensor_array.cc | 3 | ||||
-rw-r--r-- | tensorflow/core/kernels/tensor_array.h | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/tensor_array.cc b/tensorflow/core/kernels/tensor_array.cc index 765467bc1e..0e6c0ddccc 100644 --- a/tensorflow/core/kernels/tensor_array.cc +++ b/tensorflow/core/kernels/tensor_array.cc @@ -62,7 +62,8 @@ TF_CALL_complex128(TENSOR_ARRAY_WRITE_OR_ADD_GPU); } #define TENSOR_ARRAY_SET_ZERO_CPU(T) TENSOR_ARRAY_SET_ZERO(CPUDevice, T) -TF_CALL_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_CPU) +TF_CALL_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_CPU); +TF_CALL_bool(TENSOR_ARRAY_SET_ZERO_CPU); #undef TENSOR_ARRAY_SET_ZERO_CPU #if GOOGLE_CUDA diff --git a/tensorflow/core/kernels/tensor_array.h b/tensorflow/core/kernels/tensor_array.h index e8dc4fad21..384a63e945 100644 --- a/tensorflow/core/kernels/tensor_array.h +++ b/tensorflow/core/kernels/tensor_array.h @@ -81,7 +81,8 @@ Status TensorSetZero(OpKernelContext* ctx, Tensor* value) { Status TensorSetZero<Device, T>(OpKernelContext * ctx, Tensor * value); #define TENSOR_ARRAY_SET_ZERO_CPU(T) TENSOR_ARRAY_SET_ZERO(CPUDevice, T) -TF_CALL_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_CPU) +TF_CALL_NUMBER_TYPES(TENSOR_ARRAY_SET_ZERO_CPU); +TF_CALL_bool(TENSOR_ARRAY_SET_ZERO_CPU); #undef TENSOR_ARRAY_SET_ZERO_CPU #if GOOGLE_CUDA |