Home
last modified time | relevance | path

Searched refs:dstLinearIndex (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaFoldConstantTranspose.cpp73 uint64_t dstLinearIndex = dstIndices.front(); in matchAndRewrite() local
75 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
77 outputValues[dstLinearIndex] = attrValues[srcIndices]; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DConstantFold.cpp169 uint64_t dstLinearIndex = 0; in matchAndRewrite() local
196 dstLinearIndex = dstIndices.front(); in matchAndRewrite()
201 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
228 fpOutputValues[dstLinearIndex] = *computeFn(computeFnInputs).apFloat; in matchAndRewrite()
249 intOutputValues[dstLinearIndex] = *computeFn(computeFnInputs).apInt; in matchAndRewrite()