Skip to content

Commit

Permalink
chore: minor linting
Browse files Browse the repository at this point in the history
  • Loading branch information
chohk88 committed May 9, 2024
1 parent cc21afe commit 9f901f6
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions py/torch_tensorrt/dynamo/conversion/impl/normalization/ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -489,8 +489,9 @@ def cdist_forward(
- For x1.shape = [10, 5] (10 vectors, each of 5 features) and x2.shape = [20, 5] (20 vectors, each of 5 features),
since there are no batch dimensions to match, the output shape is simply [10, 20], comparing all vectors from x1 against all vectors from x2.
Note: The `compute_mode` parameter is designed to optimize the performance of the Euclidean distance calculation, especially useful when working with large datasets.
This parameter allows you to control how the distances are computed, with different modes available to leverage matrix multiplication for speed improvements.
Note: The `compute_mode` parameter is designed to optimize the performance of the Euclidean distance calculation,
especially useful when working with large datasets. This parameter allows you to control how the distances are computed,
with different modes available to leverage matrix multiplication for speed improvements.
"""
if compute_mode is None:
compute_mode = 0
Expand Down

0 comments on commit 9f901f6

Please sign in to comment.