diff options
author | 2017-06-03 18:22:52 -0700 | |
---|---|---|
committer | 2017-06-03 18:26:15 -0700 | |
commit | 7ffc3573255ecdc27b63c6d7dfefa8345225593d (patch) | |
tree | bdb68064054858fcb7656f2a8b6bc04281464558 /tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc | |
parent | aad2e3daff8fcd29ed8e5071d4c37a7f94a0421c (diff) |
Add support for bools in matrix_diag, matrix_diag_part, matrix_set_diag, matrix_band_part.
PiperOrigin-RevId: 157939272
Diffstat (limited to 'tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc b/tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc index bd097ff328..8e41ce5860 100644 --- a/tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc +++ b/tensorflow/core/kernels/matrix_set_diag_op_gpu.cu.cc @@ -29,6 +29,7 @@ typedef Eigen::GpuDevice GPUDevice; template struct functor::MatrixSetDiag<GPUDevice, T>; TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPEC); +TF_CALL_bool(DEFINE_GPU_SPEC); TF_CALL_complex64(DEFINE_GPU_SPEC); TF_CALL_complex128(DEFINE_GPU_SPEC); |