Skip to content

Conversation

@hariharans29
Copy link
Member

@hariharans29 hariharans29 commented Apr 21, 2022

Description:

The SpaceDepth ops are just thin wrappers over Transpose in that they do some re-ordering of the input based on a "virtual input shape" of the regular NCHW image input. The output tensor in these ops were being created based on the "virtual otuput shape" of this Transpose operation and then finally reshaped into the NCHW shape before the operation exits. The execution frame has a check to ensure that the requested output shape of the output buffer of the op be the same as the "expected" shape if the op output is also a graph output and if it isn't it produces a warning -

LOGS(session_state_.Logger(), WARNING) << "Expected shape from model of " << *expected_shape
. This change adjusts the op's output tensor creation logic so that such warnings don't kick in should the SpaceDepth ops be producing graph outputs.

The existing op unit tests should keep this change covered.

Motivation and Context
Fix jarring warning seen by user in #11274

@hariharans29 hariharans29 merged commit 23b0125 into master Apr 22, 2022
@hariharans29 hariharans29 deleted the hari/fixSpaceDepthCudaWarnings branch April 22, 2022 02:58
seddonm1 pushed a commit to seddonm1/onnxruntime that referenced this pull request May 15, 2022
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