coreai_opt.inspection.ModelSummary¶

class coreai_opt.inspection.ModelSummary(model, mode)[source]¶

Bases: object

Complete listing of operations discovered in a model.

Parameters:

model (ModuleInfo)
mode (ExecutionMode)

model¶

Top level module summary of the module hierarchy tree containing all discovered operations nested within their owning modules.

Type:: ModuleInfo

mode¶

Which discovery mode was used: ExecutionMode.GRAPH for exported GraphModule models, ExecutionMode.EAGER for nn.Module models traced via a forward pass.

Type:: ExecutionMode

__init__(model, mode)¶

Parameters:

model (ModuleInfo)
mode (ExecutionMode)

Return type:

None

mode: ExecutionMode¶

model: ModuleInfo¶