Skip to content

Conversation

@srkreddy1238
Copy link
Contributor

Majority of the operations made compatible with custom layouts. Incompatible ops will fallback to regular layout.

Conv1D, Conv3D, Pool1D, Pool3D, AdaptiveAvgPool1D, AdaptiveAvgPool3D are left unchanged now. 2D networks are expected to work now.

@srkreddy1238 srkreddy1238 force-pushed the convert_layout_sub_indexing branch 2 times, most recently from f8108bb to ad004ac Compare December 30, 2024 15:46
Majority of the operations made compatible with custom layouts.
Incompatible ops will fallback to regular layout.

Conv1D, Conv3D, Pool1D, Pool3D, AdaptiveAvgPool1D, AdaptiveAvgPool3D
are left unchanged now. 2D networks are expected to work now.
@srkreddy1238 srkreddy1238 force-pushed the convert_layout_sub_indexing branch from ad004ac to aef5070 Compare January 4, 2025 04:14
@srkreddy1238
Copy link
Contributor Author

@tvm-bot rerun

1 similar comment
@srkreddy1238
Copy link
Contributor Author

@tvm-bot rerun

@tqchen
Copy link
Member

tqchen commented Jan 19, 2025

@spectrometerHBH can you help to review this PR

@spectrometerHBH spectrometerHBH merged commit d641354 into apache:main Jan 19, 2025
20 checks passed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
…exing (apache#17568)

Convert layout pass and ops enhanced to support sub indexing

Majority of the operations made compatible with custom layouts.
Incompatible ops will fallback to regular layout.

Conv1D, Conv3D, Pool1D, Pool3D, AdaptiveAvgPool1D, AdaptiveAvgPool3D
are left unchanged now. 2D networks are expected to work now.
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.

3 participants