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.