polygraphy.util.format.FormatManager Class Reference

Inheritance diagram for polygraphy.util.format.FormatManager:

Collaboration diagram for polygraphy.util.format.FormatManager:

Static Public Member Functions
def	determine_format (shape)
def	permutation (old_format, new_format)
def	convert (shape, new_format)

Static Public Attributes
def	DATA_PERMUTATIONS = _generate_permutations()

Private Member Functions
def	_generate_permutations ()

Member Function Documentation

_generate_permutations()

def polygraphy.util.format.FormatManager._generate_permutations ( )

private

determine_format()

def polygraphy.util.format.FormatManager.determine_format (   shape )

static

Guesses the data format of a given shape.

Args:
    shape (Tuple[int]): The shape, including batch dimension.

Returns:
    DataFormat: The determined data format.

permutation()

def polygraphy.util.format.FormatManager.permutation (   old_format,   new_format  )

static

convert()

def polygraphy.util.format.FormatManager.convert (   shape,   new_format  )

static

Permutes a shape from one format to another.

Args:
    shape (Tuple[int]): The shape to convert.
    new_format (DataFormat): The desired format of the shape.

Returns:
    Tuple[int]: A new shape in the correct format.

Member Data Documentation

DATA_PERMUTATIONS

def polygraphy.util.format.FormatManager.DATA_PERMUTATIONS = _generate_permutations()

static

---

The documentation for this class was generated from the following file:

• format.py