Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.

Module: tfa.options

Functions

disable_custom_kernel(...): Prefer pure python operations to custom C++/CUDA kernel.

enable_custom_kernel(...): Prefer custom C++/CUDA kernel to pure python operations.

is_custom_kernel_disabled(...): Return whether custom C++/CUDA kernel is disabled.