Fix bad assert in dmlx (#142)
This commit is contained in:
Родитель
6f583b2587
Коммит
3cc9a99d88
|
@ -3591,7 +3591,7 @@ namespace dml
|
|||
}
|
||||
|
||||
TensorDesc outputROIGradientTensor = computeOutputROIGradient ? TensorDesc(roiTensor.dataType, roiTensor.sizes, builder->GetTensorPolicy()) : TensorDesc();
|
||||
assert(!computeOutputROIGradient || computeOutputROIGradient.sizes == roiTensor.sizes);
|
||||
assert(!computeOutputROIGradient || outputROIGradientTensor.sizes == roiTensor.sizes);
|
||||
|
||||
DML_ROI_ALIGN_GRAD_OPERATOR_DESC desc = {};
|
||||
desc.InputTensor = input ? inputTensor.AsPtr<DML_TENSOR_DESC>() : nullptr;
|
||||
|
|
Загрузка…
Ссылка в новой задаче