def test_tf_quantizer(self): model = TfMnist() tf_compressor.NaiveQuantizer([{ 'op_types': ['default'] }]).compress_default_graph()
def test_tf_naive_quantizer(self): tf_compressor.NaiveQuantizer(get_tf_model(), [{ 'op_types': ['default'] }]).compress()