Tf.linspace InvalidArgumentError in Mac M1 Max

Hi, I’m very new to Tensorflow on Mac M1 stuff. I was successful in installing and running tensorflow but recently I encountered an error.

I followed two tutorials. First one is to set up my TF environment.

Second one is the simple TF regression tutorial.

Some notes:

  • I updated the installation .yml file from 2.5.0 to 2.9.0 for tensorflow-deps and tensorflow-macos and 0.5.0 for tensorflow-metal
  • TF tutorial is for version 2.8.0

The error I got was an InvalidArgumentError when I tried to run tf.linspace([0.,5.],[10.,40.],5,axis=0). My Google search had been to no avail. As a last resort, I’m writing it here. Full error and a screenshot are provided below.

InvalidArgumentError: Multiple Default OpKernel registrations match NodeDef '{{node ZerosLike}}': 'op: "ZerosLike" device_type: "DEFAULT" constraint { name: "T" allowed_values { list { type: DT_INT32 } } } host_memory_arg: "y"' and 'op: "ZerosLike" device_type: "DEFAULT" constraint { name: "T" allowed_values { list { type: DT_INT32 } } } host_memory_arg: "y"' [Op:ZerosLike]

UPDATE: Error does not appear if I uninstall tensorflow-metal.

p.s. I encountered an error while editing my original post, so I had to use reply.