コード例 #1
0
 def mask_propagation(self, model: NNCFNetwork, nx_node, graph: NNCFGraph,
                      nx_graph: nx.DiGraph):
     identity_mask_propagation(nx_node, nx_graph)
     nx_node['accept_pruned_input'] = True
コード例 #2
0
ファイル: export_helpers.py プロジェクト: xiaming9880/nncf 
 def mask_propagation(self, model: NNCFNetwork, nx_node, graph: NNCFGraph,
                      nx_graph: nx.DiGraph):
     identity_mask_propagation(nx_node, nx_graph)