gigl.common.utils.tf_on_cpu

gigl.common.utils.decorator.tf_on_cpu(func: Callable[[...], _ReturnType]) → Callable[[...], _ReturnType]

A decorator to run a function using TensorFlow’s CPU device.