Skip to content

[Torch] Support aten::tensor, aten::empty and aten::numel #6535

Description

@zhiqwang

Hi TVM teams,

TVM currently support pytorch 1.4 and torchvision 0.5. I want to know is there any plan to support pytorch 1.6 version to trace the object detection models in pytorch 1.6?

For faster rcnn, which is not partly scripted and thus can be traced completely, I get following missing ops with PyTorch 1.6

NotImplementedError: The following operators are not implemented: ['aten::tensor', 'aten::empty', 'aten::numel']

Originally posted by @masahi in #6449 (comment)

Activity

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

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions