Skip to content

Commit

Permalink
chore: minor linting
Browse files Browse the repository at this point in the history
  • Loading branch information
chohk88 committed Apr 29, 2024
1 parent e359059 commit 1962cf8
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 @@ -551,8 +551,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 1962cf8

Please sign in to comment.