tensorflow-metal plugin problem with grouped convolutions

Running grouped convolutions on an M2 with the metal plugin I get an error. Example code:

Using TF2.11 and no metal plugin I get

import tensorflow as tf
tf.keras.layers.Conv1D(5,1,padding="same", kernel_initializer="ones", groups=5)(tf.ones((1,1,5)))
# displays
<tf.Tensor: shape=(1, 1, 5), dtype=float32, numpy=array([[[1., 1., 1., 1., 1.]]], dtype=float32)>

On TF2.14 with the plugin I received

import tensorflow as tf
tf.keras.layers.Conv1D(5,1,padding="same", kernel_initializer="ones", groups=5)(tf.ones((1,1,5)))

# displays
...

NotFoundError: Exception encountered when calling layer 'conv1d_3' (type Conv1D).

could not find registered platform with id: 0x104d8f6f0 [Op:__inference__jit_compiled_convolution_op_78]

Call arguments received by layer 'conv1d_3' (type Conv1D):
  • inputs=tf.Tensor(shape=(1, 1, 5), dtype=float32)

could not find registered platform with id

tensorflow-metal plugin problem with grouped convolutions
 
 
Q