Missing GPU implementation Op:StatelessRandomGetKeyCounter for the Embedding layer in tensorflow-metal

The Keras Embedding layer cannot be calculated on Metal because of the missing Op:StatelessRandomGetKeyCounter, as shown in this error message:

tensorflow.python.framework.errors_impl.InvalidArgumentError: Could not satisfy device specification '/job:localhost/replica:0/task:0/device:GPU:0'. enable_soft_placement=0. Supported device types [CPU]. All available devices [/job:localhost/replica:0/task:0/device:GPU:0, /job:localhost/replica:0/task:0/device:CPU:0]. [Op:StatelessRandomGetKeyCounter]

A workaround is to enable soft placement, but this obviously is slower:

tf.config.set_soft_device_placement(True)

Reporting it here as recommended by the TensorFlow Plugin Metal team.

Missing GPU implementation Op:StatelessRandomGetKeyCounter for the Embedding layer in tensorflow-metal
 
 
Q