coreai_opt.inspection.BoundaryEdge¶

class coreai_opt.inspection.BoundaryEdge(op, index)[source]¶

Bases: object

A single data-flow edge crossing a module boundary.

Each entry in ModuleInfo.input_ops or ModuleInfo.output_ops corresponds to one quantizer configuration point at the boundary.

Parameters:

op (OpInfo)
index (int)

op¶

The op inside the module at the boundary.

Type:: OpInfo

index¶

For input boundaries: the input slot of op that receives external data, matching the index used in module_input_spec. For output boundaries: the output slot of op whose tensor leaves the module, matching the index used in module_output_spec.

Type:: int

__init__(op, index)¶

Parameters:

op (OpInfo)
index (int)

Return type:

None

op: OpInfo¶