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

Fix 5d concat operator helper

Browse source
This commit is contained in:
jake mathern 2020-05-19 15:07:33 -07:00
parent f57515aae4
commit 2adb0f8d1d
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();