ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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.