gigl.common.utils.decorator#

Functions#

tf_on_cpu(func)

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

Module Contents#

gigl.common.utils.decorator.tf_on_cpu(func)[source]#

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

Parameters:: func (Callable[Ellipsis, _ReturnType])
Return type:: Callable[Ellipsis, _ReturnType]