已合并
【DW】增加dw转matmul的format拦截 #9452
yejiani创建于 8月28日
【DW】增加dw转matmul的format拦截 #9452
已合并
共 1 个文件变更+5-0
| @@ -2349,6 +2349,11 @@ static bool Is1x1DwTransToMm(const ConvolutionBackwardInputTensor& inputTensor, | |||
| 2349 | } | 2349 | } |
| 2350 | op::Shape weightShape = inputTensor.weight->GetViewShape(); | 2350 | op::Shape weightShape = inputTensor.weight->GetViewShape(); |
| 2351 | // 5D NCDHW weight + 1x1x1 kernel | 2351 | // 5D NCDHW weight + 1x1x1 kernel |
 | |||
| 2352 | + if (inputTensor.gradOutput->GetStorageFormat() != op::Format::FORMAT_NCDHW || | ||
| 2353 | + inputTensor.input->GetStorageFormat() != op::Format::FORMAT_NCDHW || | ||
| 2354 | + inputTensor.weight->GetStorageFormat() != op::Format::FORMAT_NCDHW) { | ||
| 2355 | + return false; | ||
| 2356 | + } | ||
| 2352 | if (weightShape.GetDimNum() != CONV3DINPUTDIM || weightShape[dDimNCDHWIdx] != 1 || weightShape[hDimNCDHWIdx] != 1 || | 2357 | if (weightShape.GetDimNum() != CONV3DINPUTDIM || weightShape[dDimNCDHWIdx] != 1 || weightShape[hDimNCDHWIdx] != 1 || |
| 2353 | weightShape[wDimNCDHWIdx] != 1) { | 2358 | weightShape[wDimNCDHWIdx] != 1) { |
| 2354 | return false; | 2359 | return false; |
注释的位置放的不太合适