colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add comment.

This commit is contained in:
Doru Bercea 2020-01-22 15:05:56 -05:00
parent ea45cbcca9
commit 94391a3cde
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/dialect/onnx/onnx_ops.cpp
View File

@ -484,6 +484,9 @@ void ONNXConvNoBiasOp::inferShapes() {
if (dataShape[1] != (weightShape[1] * group)) if (dataShape[1] != (weightShape[1] * group))
emitError("Channel dimension mismatch."); emitError("Channel dimension mismatch.");
// Note: the value of the group attribut only impacts the way the
// computation is carried out and not the actual output size.
// First two output dimensions consist of the number of batches and the // First two output dimensions consist of the number of batches and the
// number of kernels being applied. // number of kernels being applied.
// //