Lines Matching refs:setShapeInfo
490 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo() function in __anon8ba1aee70111::LowerMatrixIntrinsics
574 Propagate = setShapeInfo(Inst, {M, K}); in propagateShapeForward()
578 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
582 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
586 Propagate = setShapeInfo(Inst, {M, N}); in propagateShapeForward()
590 setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
597 Propagate |= 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()
785 setShapeInfo(M, {C, R}); in optimizeTransposes()