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.