Skip to content

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

@narendasan

Description

@narendasan

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

Doesnt TensorRT support INT64 I/O now?

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions