polygraphy.comparator.data_loader.BaseDataLoader Class Reference

Inheritance diagram for polygraphy.comparator.data_loader.BaseDataLoader:

Collaboration diagram for polygraphy.comparator.data_loader.BaseDataLoader:

Public Member Functions
def	__getitem__ (self, index)

Detailed Description

Responsible for fetching or generating input data for runners.

Member Function Documentation

__getitem__()

def polygraphy.comparator.data_loader.BaseDataLoader.__getitem__	(		self,
			index
	)

Fetches or generates inputs.

Args:
    index (int): The index of inputs to fetch. For any given index, the inputs should always be the same.

Vars:
    input_metadata (TensorMetadata):
    Mapping of input names to their data types and shapes.
    This is set by the Comparator when the data loader is used.

Returns:
    OrderedDict[str, numpy.ndarray]: Mapping of input names to NumPy buffers containing data.

Reimplemented in polygraphy.comparator.data_loader.DataLoader.

---

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

• comparator/data_loader.py