Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Relaxing convolution infer checks. #3511

Merged
merged 1 commit into from
Jul 9, 2019

Conversation

anijain2305
Copy link
Contributor

  • Weight dtype can be different than idtype. So, using the weight tensor to set
    the dtype of weight.
  • For conv2d NCHWc operator, the weight can be of any dimension. For int8
    computation on Intel, it can be 7D. Relaxing the weight type checking.

Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers.

@anijain2305
Copy link
Contributor Author

@yzhliu @kevinthesun @tqchen @rankyung-hong Please review.

- Weight dtype can be different than idtype. So, using the weight tensor to set
the dtype of weight.
- For conv2d NCHWc operator, the weight can be of any dimension. For int8
computation on Intel, it can be 7D. Relaxing the weight type checking.
@yzhliu yzhliu merged commit 3cab3c4 into apache:master Jul 9, 2019
@yzhliu
Copy link
Member

yzhliu commented Jul 9, 2019

Thanks @anijain2305 !

@anijain2305 anijain2305 deleted the conv_infer_relax branch July 9, 2019 17:16
wweic pushed a commit to wweic/tvm that referenced this pull request Jul 11, 2019
- Weight dtype can be different than idtype. So, using the weight tensor to set
the dtype of weight.
- For conv2d NCHWc operator, the weight can be of any dimension. For int8
computation on Intel, it can be 7D. Relaxing the weight type checking.
wweic pushed a commit to wweic/tvm that referenced this pull request Jul 11, 2019
- Weight dtype can be different than idtype. So, using the weight tensor to set
the dtype of weight.
- For conv2d NCHWc operator, the weight can be of any dimension. For int8
computation on Intel, it can be 7D. Relaxing the weight type checking.
wweic pushed a commit to neo-ai/tvm that referenced this pull request Jul 11, 2019
- Weight dtype can be different than idtype. So, using the weight tensor to set
the dtype of weight.
- For conv2d NCHWc operator, the weight can be of any dimension. For int8
computation on Intel, it can be 7D. Relaxing the weight type checking.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants