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

DML EP squeeze all axes when empty (#12649)

Browse source 
DML EP squeeze empty axes
This commit is contained in:
Dwayne Robinson 2022-08-19 08:56:03 -07:00 committed by GitHub
parent b270334e1e
commit aa85092b51
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1852,8 +1852,14 @@ namespace OperatorHelper
m_axes = kernelInformation.GetAttributes().GetOptionalAttributeVectorInt32(AttrName::Axes);
}
std::vector<DimensionType> inputDimensions = shapeInformation.GetInputTensorShape(0);
HandleNegativeAxes(/*inout*/ m_axes, gsl::narrow_cast<uint32_t>(inputDimensions.size()));
std::sort(m_axes.begin(), m_axes.end());
if (m_axes.empty())
{
m_axes.resize(inputDimensions.size());
std::iota(m_axes.begin(), m_axes.end(), 0u);
}
}
std::vector<EdgeShapes> SqueezeHelper::GetOutputShapes(const MLShapeInferenceContext& shapeInfo) const