Skip to content

Commit

Permalink
add bfloat16 support to paddle backend broadcast_to
Browse files Browse the repository at this point in the history
  • Loading branch information
MahmoudAshraf97 committed Jul 23, 2023
1 parent 69a44f0 commit 999ab2d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ivy/functional/backends/paddle/data_type.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,7 @@ def broadcast_to(
paddle.int16,
paddle.uint8,
paddle.float16,
paddle.bfloat16,
]:
return paddle.broadcast_to(x.cast("float32"), shape).cast(x.dtype)
elif x.dtype in [paddle.complex64, paddle.complex128]:
Expand Down

0 comments on commit 999ab2d

Please sign in to comment.