Skip to content

Commit

Permalink
Fix quantization
Browse files Browse the repository at this point in the history
  • Loading branch information
vinx13 committed Apr 1, 2019
1 parent c7a51e2 commit 190943a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/relay/quantize/_annotate.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def conv2d_rewrite(ref_call, new_args, ctx):
def dense_rewrite(ref_call, new_args, ctx):
"""Rewrite function for dense. Lhs of dense will be quantized to input field, and rhs of
dense will be quantized to weight field. Output would be in activation field."""
if not current_qconfig.quantize_dense:
if not current_qconfig().quantize_dense:
return None
cnt = _conv_counter()
if cnt < current_qconfig().skip_k_conv:
Expand Down

0 comments on commit 190943a

Please sign in to comment.