Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit d27ed9c

Browse files
nolanlioujhseu
authored andcommitted
Support boolean type to variables' ops and assign-like ops.(tensorflow#12046) (tensorflow#12292)
* Support boolean type to variable and assign-like ops. * Remove boolean type to assign add and sub. * Remove duplicate initialize kernel of dense_update_ops.
1 parent 27b9361 commit d27ed9c

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tensorflow/core/kernels/variable_ops.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
8383
IsVariableInitializedOp);
8484

8585
TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS);
86+
TF_CALL_bool(REGISTER_GPU_KERNELS)
8687
#undef REGISTER_GPU_KERNELS
8788
#endif // GOOGLE_CUDA
8889

0 commit comments

Comments
 (0)