Searched refs:setShapeInfo (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/llvm/lib/Transforms/Scalar/ |
| H A D | LowerMatrixIntrinsics.cpp | 574 Propagate = setShapeInfo(Inst, {M, K}); in propagateShapeForward()
578 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
590 setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
645 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
648 if (setShapeInfo(MatrixB, {N, K})) in propagateShapeBackward()
654 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
659 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
672 if (setShapeInfo(U.get(), Shape)) in propagateShapeBackward()
749 setShapeInfo(T0, {C, K}); in optimizeTransposes()
753 setShapeInfo(T1, {K, R}); in optimizeTransposes()
[all …]
|