def test_tf_pruner(self): model = TfMnist() configure_list = [{'sparsity': 0.8, 'op_types': ['default']}] tf_compressor.LevelPruner(configure_list).compress_default_graph()
def test_tf_level_pruner(self): configure_list = [{'sparsity': 0.8, 'op_types': ['default']}] tf_compressor.LevelPruner(get_tf_model(), configure_list).compress()