Skip to content

Commit 283c622

Browse files
author
Ryan Lai
authored
Remove artifical limitation on 5D tensors (#325)
1 parent 929dac9 commit 283c622

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

Tools/WinMLRunner/src/Run.cpp

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -293,13 +293,6 @@ HRESULT CheckIfModelAndConfigurationsAreSupported(LearningModel& model, const st
293293
else if (inputFeature.Kind() == LearningModelFeatureKind::Tensor)
294294
{
295295
auto tensorFeatureDescriptor = inputFeature.try_as<TensorFeatureDescriptor>();
296-
if (tensorFeatureDescriptor.Shape().Size() > 4 && deviceType != DeviceType::CPU)
297-
{
298-
std::cout << "Input feature " << to_string(inputFeature.Name())
299-
<< " shape is too large. GPU path only accepts tensor dimensions <= 4 : "
300-
<< tensorFeatureDescriptor.Shape().Size() << std::endl;
301-
return E_INVALIDARG;
302-
}
303296

304297
// If image as input binding, then the model's tensor inputs should have channel 3 or 1
305298
if (hasInputBindingImage &&

0 commit comments

Comments
 (0)