neural_compressor.ux.components.graph.graph_reader

Graph reader.

Module Contents

Classes

GraphReader

Model graph reader.

class neural_compressor.ux.components.graph.graph_reader.GraphReader

Model graph reader.

read(model_path: str, expanded_groups: List[str]) → neural_compressor.ux.components.graph.graph.Graph

Return Graph for given model path.

find_pattern_in_graph(model_path: str, op_name: str, pattern: List[str]) → Tuple[neural_compressor.ux.components.graph.graph.Graph, List[str]]

Search graph for specific nodes pattern.