Searched refs:cSize (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 1385 int64_t nSize, wSize, cSize, kwSize, fSize; in conv() local
1387 bindShapeDims(rhsShapedType, kwSize, cSize, fSize); in conv()
1408 cSize}, in conv()
1435 /*sizes=*/ArrayRef<int64_t>{nSize, wSizeStep, cSize}, in conv()
1509 int64_t nSize, wSize, cSize, kwSize; in depthwiseConv() local
1511 bindShapeDims(rhsShapedType, kwSize, cSize); in depthwiseConv()
1531 cSize}, in depthwiseConv()
1533 VectorType rhsType = VectorType::get({kwSize, cSize}, rhsEltType); in depthwiseConv()
1534 VectorType resType = VectorType::get({nSize, wSize, cSize}, resEltType); in depthwiseConv()
1559 /*sizes=*/ArrayRef<int64_t>{nSize, wSizeStep, cSize}, in depthwiseConv()
[all …]
|