pytorch_nndct.InputSpec - 1.3 English

Vitis AI Optimizer User Guide (UG1333)

Document ID: UG1333
Release Date: 2020-12-17
Version: 1.3 English

Specifies the dtype and shape of every input to a module.

Arguments

InputSpec(shape, dtype)

shape: Shape tuple, expected shape of the input.
dtype: Expected torch.dtype of the input.