Deeploy.DeeployTypes.NetworkOptimizationPass

class Deeploy.DeeployTypes.NetworkOptimizationPass

Bases: TopologyOptimizationPass

Pass to update the NetworkContext and Neural Network Graph in one go

Methods

__init__()

__init__()
apply(ctxt, graph)	The method to update context and graph

apply(ctxt: NetworkContext, graph: Graph) → Tuple[NetworkContext, Graph]

The method to update context and graph

Parameters:

• ctxt (NetworkContext) – Current NetworkContext

• graph (gs.Graph) – Current Neural Network graph

Returns:

Updated context and graph

Return type:

Tuple[NetworkContext, gs.Graph]