Getting the same error when working with 64-bit float variables. Minimal example:
import tensorflow as tf
DTYPE = "float64"
def mre():
P = tf.Variable(tf.ones(shape=(), dtype=DTYPE))
delta = tf.ones(dtype=DTYPE, shape=P.shape)
print(f"{P.numpy()=}")
print(f"{P.dtype=}")
P.assign_add(delta) # crashes here
print(P.numpy())
if __name__ == "__main__":
mre()
Environment:
Python 3.11.7
tensorflow 2.15.0
tensorflow-estimator 2.15.0
tensorflow-io-gcs-filesystem 0.36.0
tensorflow-macos 2.15.0
tensorflow-metal 1.1.0
Chip: M3 Pro
OS: 14.4.1 (23E224)
SiLiKhon
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution
Last seen
User for