Skip to content

[aten::embedding] Do we still need this check? #4008

Description

@narendasan

if isinstance(indices_tensor, torch.Tensor) and indices_tensor.dtype == torch.int64:

Doesnt TensorRT support INT64 I/O now?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

story: Operator Coverage & ConvertersMissing ATen op converters, converter accuracy bugs, type-casting issues, and new operator support r

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions