Can't customize a TensorFlow operation

Following this video "Accelerate machine learning with Metal" & this sample code "Customizing a TensorFlow operation"

My Environment:

  • macOS 13.1
  • Python 3.8.15
  • tensorflow-deps 2.9.0
  • tensorflow-macos 2.9.0
  • tensorflow-metal 0.5.1

after running make at hash_encoder, the hash_encoder_kernel.so appears. Then I tried to run python tiny_nerf_hash.py but it didn't work.

Unlike the python tiny_nerf_mlp.py, it works normally but it didn't take the advantage of accelerating.

Can't customize a TensorFlow operation
 
 
Q