saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-27 03:11:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merged PR 4700287: Fix 5d concat operator helper

Browse source 
there is an error check in the concat helper that prevents 5d from working

Related work items: #21113572
This commit is contained in:
Jake Mathern 2020-05-19 22:11:23 +00:00
commit 4e84df25e7
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
View file

@ -971,7 +971,7 @@ namespace OperatorHelper
std::vector<EdgeShapes> ConcatHelper::GetOutputShapes(const MLShapeInferenceContext& shapeInfo) const
{
auto outputShape = shapeInfo.GetInputTensorShape(0);
ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NchwDimensionCount);
ML_CHECK_VALID_ARGUMENT(outputShape.size() <= NcdhwDimensionCount);
uint32_t inputCount = shapeInfo.GetInputCount();