polygraphy.comparator.data_loader.DataLoaderCache Class Reference

Inheritance diagram for polygraphy.comparator.data_loader.DataLoaderCache:

Collaboration diagram for polygraphy.comparator.data_loader.DataLoaderCache:

Public Member Functions
def	__init__ (self, data_loader, save_inputs_path=None)
def	__getitem__ (self, iteration)
def	set_input_metadata (self, input_metadata)

Public Attributes
	data_loader
	cache
	save_inputs_path
	input_metadata

Constructor & Destructor Documentation

__init__()

def polygraphy.comparator.data_loader.DataLoaderCache.__init__	(		self,
			data_loader,
			save_inputs_path = None
	)

Member Function Documentation

__getitem__()

def polygraphy.comparator.data_loader.DataLoaderCache.__getitem__	(		self,
			iteration
	)

Load the specified iteration from the cache if present, or generate using the data loader.

Args:
    iteration (int): The iteration whose data to retrieve.

set_input_metadata()

def polygraphy.comparator.data_loader.DataLoaderCache.set_input_metadata	(		self,
			input_metadata
	)

Set the input metadata for the data loader.

Args:
    input_metadata (TensorMetadata):
    Input Metadata, including shape and type information. The cache may attempt to transform inputs to
    match the specified input_metadata when data already in the cache does not exactly match.

Member Data Documentation

data_loader

polygraphy.comparator.data_loader.DataLoaderCache.data_loader

cache

polygraphy.comparator.data_loader.DataLoaderCache.cache

save_inputs_path

polygraphy.comparator.data_loader.DataLoaderCache.save_inputs_path

input_metadata

polygraphy.comparator.data_loader.DataLoaderCache.input_metadata

---

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

• comparator/data_loader.py