Skip to content

pixano_inference.frameworks.tensorflow

TensorFlow framework adapter.

Provides framework-neutral device/dtype resolution and numpy interchange for custom models written in TensorFlow. TensorFlow is imported lazily; importing this module does not require it.

TensorFlowAdapter

:class:~pixano_inference.frameworks.base.FrameworkAdapter for TensorFlow.

from_numpy(array, device=None, dtype=None)

Convert a numpy array to a TensorFlow tensor on device with dtype.

Source code in pixano_inference/frameworks/tensorflow.py 
55
56
57
58
59
60
61
62
def from_numpy(self, array: np.ndarray, device: Any = None, dtype: Any = None) -> Any:
    """Convert a numpy array to a TensorFlow tensor on *device* with *dtype*."""
    import tensorflow as tf

    if device is not None:
        with tf.device(device):
            return tf.convert_to_tensor(array, dtype=dtype)
    return tf.convert_to_tensor(array, dtype=dtype)

is_available()

Whether tensorflow is importable.

Source code in pixano_inference/frameworks/tensorflow.py 
30
31
32
def is_available(self) -> bool:
    """Whether tensorflow is importable."""
    return is_tensorflow_installed()

resolve_device(num_gpus)

Return a TensorFlow device string ("/GPU:0" when requested and present).

Source code in pixano_inference/frameworks/tensorflow.py 
34
35
36
37
38
39
40
def resolve_device(self, num_gpus: float) -> Any:
    """Return a TensorFlow device string (``"/GPU:0"`` when requested and present)."""
    import tensorflow as tf

    if num_gpus > 0 and tf.config.list_physical_devices("GPU"):
        return "/GPU:0"
    return "/CPU:0"

resolve_dtype(dtype)

Resolve a tf.DType from a string (e.g. "float32", "bfloat16").

Source code in pixano_inference/frameworks/tensorflow.py 
42
43
44
45
46
47
48
49
def resolve_dtype(self, dtype: str) -> Any:
    """Resolve a tf.DType from a string (e.g. ``"float32"``, ``"bfloat16"``)."""
    import tensorflow as tf

    try:
        return tf.dtypes.as_dtype(dtype)
    except TypeError as exc:
        raise ValueError(f"Unsupported tensorflow dtype '{dtype}'.") from exc

to_numpy(array)

Convert a TensorFlow tensor to a numpy array.

Source code in pixano_inference/frameworks/tensorflow.py 
51
52
53
def to_numpy(self, array: Any) -> np.ndarray:
    """Convert a TensorFlow tensor to a numpy array."""
    return np.asarray(array)