tfa.options.enable_custom_kernel

Stay organized with collections Save and categorize content based on your preferences.

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.