Find it here: https://developer.apple.com/documentation/metal/metal_sample_code_library/customizing_a_tensorflow_operation