Skip to content

Enhancement for fold_scale_axis and dnnl_json_runtime#11815

Merged
masahi merged 3 commits into
apache:mainfrom
crazydemo:upstream-enhancement
Jul 4, 2022
Merged

Enhancement for fold_scale_axis and dnnl_json_runtime#11815
masahi merged 3 commits into
apache:mainfrom
crazydemo:upstream-enhancement

Conversation

@crazydemo
Copy link
Copy Markdown

This PR modifies two files.

  1. add missing param checks for conv's fold_scale_axis.
  2. enhance WA in dnnl_convolution, more axises need to be checked when crop weight tensors.

@crazydemo crazydemo force-pushed the upstream-enhancement branch from 8936116 to 043856e Compare July 4, 2022 02:02
@crazydemo
Copy link
Copy Markdown
Author

@masahi Could you please review this PR? It only contains some bug fix.

@masahi masahi merged commit 5efe8b0 into apache:main Jul 4, 2022
blackkker pushed a commit to blackkker/tvm that referenced this pull request Jul 7, 2022
* enhance WA in dnnl_convolution, support crop for tensor with mismatched groups and OC

* add missing param checks for conv2d, conv3d

* fix lint
masahi pushed a commit to masahi/tvm that referenced this pull request Jul 15, 2022
* enhance WA in dnnl_convolution, support crop for tensor with mismatched groups and OC

* add missing param checks for conv2d, conv3d

* fix lint
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
* enhance WA in dnnl_convolution, support crop for tensor with mismatched groups and OC

* add missing param checks for conv2d, conv3d

* fix lint
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