Deeploy.MemoryLevelExtension.OptimizationPasses.MemoryLevelAnnotationPasses.AnnotateNeurekaWeightMemoryLevel
- class Deeploy.MemoryLevelExtension.OptimizationPasses.MemoryLevelAnnotationPasses.AnnotateNeurekaWeightMemoryLevel(neurekaEngineName: str, weightMemoryLevel: MemoryLevel)
Bases:
SequentialPass
Methods
- __init__(neurekaEngineName: str, weightMemoryLevel: MemoryLevel)
__init__
(neurekaEngineName, weightMemoryLevel)apply
(ctxt, graph)named_subpasses
()register_subpass
(name, value)remove_subpass
(name)setup_passes
(passes)