tfa.options.disable_custom_kernel

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

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

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