Deeploy.EngineExtension.OptimizationPasses.TopologyOptimizationPasses.EngineColoringPasses.EngineColoringPass

class Deeploy.EngineExtension.OptimizationPasses.TopologyOptimizationPasses.EngineColoringPasses.EngineColoringPass(engineMapper: EngineMapper)

Bases: TopologyOptimizationPass

Methods

__init__(engineMapper: EngineMapper)

__init__(engineMapper)
apply(graph)	Applies a transformation to a graph

apply(graph: Graph) → Tuple[Graph]

Applies a transformation to a graph

Parameters:

graph (gs.Graph) – The neural network being deployed

Returns:

A modified version of the neural network graph

Return type:

Tuple[gs.Graph]