masahi commented on a change in pull request #5204: [Frontend][Torch] Fix up
graph input handling
URL: https://github.com/apache/incubator-tvm/pull/5204#discussion_r401744457
##########
File path: python/tvm/relay/frontend/qnn_torch.py
##########
@@ -101,20 +101,19 @@ def get_weight_quant_params(script_module):
return quant_params
-def add_quant_params_to_outputs(outputs, output_index_map,
- packed_param_map, quant_params):
+def add_quant_params_to_inputs(input_vars, packed_param_map,
+ quant_params):
"""
- Add quant params to outputs so that they can be referenced by other
+ Add quant params to inputs so that they can be referenced by other
ops later. Weights are quantized here.
Review comment:
For L104 and L107, please keep `outputs`
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
