TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

`aten._cdist_forward`

Open chohk88 opened this issue 1 year ago • 0 comments

Implement a converter for aten._cdist_forward

https://pytorch.org/docs/stable/generated/torch.cdist.html#torch.cdist https://github.com/pytorch/pytorch/blob/f00ece024bd70e6a5f3aa763715a1f353cfdacdf/aten/src/ATen/native/Distance.cpp#L81

chohk88 avatar Apr 04 '24 08:04 chohk88