The number of GPUs available to execute operations.
(self)
| 961 | self._thread_local_data.function_call_options = options |
| 962 | |
| 963 | def num_gpus(self): |
| 964 | """The number of GPUs available to execute operations.""" |
| 965 | self.ensure_initialized() |
| 966 | return self._num_gpus |
| 967 | |
| 968 | def add_function(self, fn): |
| 969 | """Add a function definition to the context. |