neural_compressor.ux.components.graph.collapser

Graph collapser.

Module Contents

Classes

Collapser

Graph collapser.

class neural_compressor.ux.components.graph.collapser.Collapser(expanded_groups: List[str] = [])

Graph collapser.

collapse(graph: neural_compressor.ux.components.graph.graph.Graph) neural_compressor.ux.components.graph.graph.Graph

Return graph with collapsed nodes.