tfa.options.enable_custom_kernel

Prefer custom C++/CUDA kernel to pure python operations.

Enable using custom C++/CUDA kernel instead of pure python operations. It has the same effect as setting environment variable TF_ADDONS_PY_OPS=0.