Skip to content

enable conv transpose 3D - #4218

Merged
Ashwini Khade (askhade) merged 2 commits into
masterfrom
askhade/add_convtranspose_3D
Jun 15, 2020
Merged

enable conv transpose 3D#4218
Ashwini Khade (askhade) merged 2 commits into
masterfrom
askhade/add_convtranspose_3D

Conversation

@askhade

Copy link
Copy Markdown
Contributor

Description: Enabling support for ConvTranspose 3D for CPU. Earlier only ConvTranspose 1D and 2D were supported.

PR (PR: #2578 ) for adding 1D ConvTranspose had already generalized the code to also support 3D conv transpose but at that time there was no test data available to thoroughly test this. This PR removes the check to disable 3D conv transpose and adds tests for same.

Motivation and Context

  • Required to serve u-Net 3D model

@askhade
Ashwini Khade (askhade) requested a review from a team as a code owner June 12, 2020 23:04
@askhade
Ashwini Khade (askhade) merged commit 4486c66 into master Jun 15, 2020
@askhade
Ashwini Khade (askhade) deleted the askhade/add_convtranspose_3D branch June 15, 2020 20:38
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