Skip to content

Commit

Permalink
remove comment
Browse files Browse the repository at this point in the history
  • Loading branch information
elad-c committed Sep 25, 2024
1 parent 0a228f5 commit a4156be
Showing 1 changed file with 0 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,6 @@ def get_tpc(self):

tpc = tp.TargetPlatformCapabilities(tp_model)
with tpc:
# No need to quantize Flatten and Dropout layers
tp.OperationsSetToLayers("WeightQuant", [torch.Tensor.expand, torch.cat])

return tpc
Expand Down

0 comments on commit a4156be

Please sign in to comment.