↔ [Converter] Add support for aten::atan2 in Torch-TensorRT #938
Labels
component: converters
Issues re: Specific op converters
component: partitioning
feature request
New feature or request
No Activity
Unsupported operator: aten::atan2(Tensor self, Tensor other) -> (Tensor)
Function Schema:
aten::atan2(Tensor self, Tensor other) -> (Tensor)
Original PyTorch API:
torch.atan2(input, other, *, out=None) → [Tensor]
Relevant TensorRT Documentation:
None
Alternatives
Additional context
When using Fourier transformation in audio segmentation models,
torch.atan2
will be used.Because variable audio input lengths requires dynamic shape, fallback to pytorch would cause runtime error when creating example inputs by
partitioning::generateRandomInputs
.It reports
Therefore, we need convertor for
torch.atan2
to successfully convert such models.The text was updated successfully, but these errors were encountered: