This commit is contained in:
Patrice Vignola 2021-07-23 11:46:13 -07:00 коммит произвёл GitHub
Родитель 6f583b2587
Коммит 3cc9a99d88
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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;