polygraphy.common.struct.TensorMetadata Class Reference

Inheritance diagram for polygraphy.common.struct.TensorMetadata:

Collaboration diagram for polygraphy.common.struct.TensorMetadata:

Public Member Functions
def	add (self, name, dtype, shape)
def	__repr__ (self)
def	__str__ (self)

Detailed Description

An OrderedDict[str, Tuple[numpy.dtype, Tuple[int]]] that maps input names to their data types and shapes.

Member Function Documentation

add()

def polygraphy.common.struct.TensorMetadata.add	(		self,
			name,
			dtype,
			shape
	)

Convenience function for adding entries.

Args:
    name (str): The name of the input.
    dtype (numpy.dtype): The data type of the input.
    shape (Tuple[int]):
    The shape of the input. Dynamic dimensions may
    be indicated by negative values, ``None``, or a string.

Returns:
    The newly added entry.

__repr__()

def polygraphy.common.struct.TensorMetadata.__repr__

(

self

)

Here is the call graph for this function:

__str__()

def polygraphy.common.struct.TensorMetadata.__str__

(

self

)

Here is the call graph for this function:

---

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

• common/struct.py