tfmindi.set_backend

Contents

tfmindi.set_backend#

tfmindi.set_backend(backend)#

Explicitly set the computational backend.

Parameters:

backend (Literal['cpu', 'gpu']) – Backend type: “cpu” or “gpu”

Raises:
  • ValueError – If backend is not supported

  • ImportError – If GPU backend is requested but packages are not available

Return type:

None