Searched refs:createTensorDim (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Conversion/TosaToSCF/ |
| H A D | TosaToSCF.cpp | 86 static Value createTensorDim(OpBuilder &builder, Location loc, Value tensor, in createTensorDim() function in __anon3cadb22c0111::ScatterOpConverter
107 auto dimN = createTensorDim(rewriter, loc, input, 0); in matchAndRewrite()
108 auto dimW = createTensorDim(rewriter, loc, input, 1); in matchAndRewrite()
109 auto dimC = createTensorDim(rewriter, loc, input, 2); in matchAndRewrite()
|