Searched refs:outputChannels (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeTransposeConv.cpp | 187 int64_t outputChannels = weightTy.getDimSize(0); in matchAndRewrite() local
222 outputChannels, weightHeight / stride[0], in matchAndRewrite()
240 outputChannels * stride[0] * stride[1], weightHeight / stride[0], in matchAndRewrite()
282 RankedTensorType::get({outputChannels * stride[0] * stride[1]}, in matchAndRewrite()
285 RankedTensorType::get({outputChannels * stride[0] * stride[1]}, in matchAndRewrite()
319 batch, convHeight, convWidth, stride[0], stride[1], outputChannels}; in matchAndRewrite()
335 batch, convHeight * stride[0], convWidth * stride[1], outputChannels}; in matchAndRewrite()
|