rc.task.contexts§

Context managers

Methods§

Timer([name, is_inline])

Context Manager for timing operations.

Environment([name, device])

Context Manager setting up the environment to task operations.

Module Contents§

Timer(name='', is_inline=True)§

Context Manager for timing operations.

Parameters:

  • name (str) – The name of this context, ``print``ed as what is being timed. The (default) empty string will not be timed.

  • is_inline (bool) – Whether to report timing inline (the default), or with linebreaks to top and tail a paragraph.

Environment(name='', device='', **kwargs)[source]§

Context Manager setting up the environment to task operations.

Parameters:

  • name (str) – The name of this context, ``print``ed as what is being task. The (default) empty string will not be timed.

  • device (str) – The device to task on. If this ends in the regex [C,G]PU* then the logical device /[C,G]PU* is used, otherwise device allocation is automatic.

  • **kwargs – Is passed straight to the implementation GPFlow manager. Note, however, that float=float32 is inoperative due to SciPy. eager=bool is passed to tf.config.run_functions_eagerly.