habana-tensorflow 1.4.0.442

This package contains Habana® TensorFlow bridge that allows to work with Habana® Gaudi®

Habana TensorFlow Python package

habana-tensorflow is a package designed for easy integration of Gaudi bridge to Tensorflow topologies. Main entry point (Python import) is habana_frameworks.tensorflow module that contains all the necessary functions to work with Gaudi.

Structure

Properly built wheel contains:

• habana_frameworks python namespace (with all the folder structure inside)
• libraries for all supported versions of TF (see habana_frameworks.tensorflow.sysconfig._supported_tf_versions) located in habana_frameworks/tensorflow/tf<version_with_underscores>/lib/habanalabs/
• proper licensing

When loading habana_tensorflow, there is sanity check for currently installed TF version in the system, to match libraries found in the module.

TensorFlow package (habana_frameworks.tensorflow)

Most important module inside is library_loader and function load_habana_module(), that is exposed directly after import. It initializes the module properly.

Example:

import habana_frameworks.tensorflow as htf
htf.load_habana_module()

List of modules inside habana_frameworks.tensorflow

All modules are described in official docs.