Lines Matching refs:setShapeInfo
529 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo() function in __anon04e69bad0111::LowerMatrixIntrinsics
622 Propagate = setShapeInfo(Inst, {M, K}); in propagateShapeForward()
626 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
630 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
634 Propagate = setShapeInfo(Inst, {M, N}); in propagateShapeForward()
638 setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
645 Propagate |= setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
693 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
696 if (setShapeInfo(MatrixB, {N, K})) in propagateShapeBackward()
702 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
707 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
720 if (setShapeInfo(U.get(), Shape)) in propagateShapeBackward()
747 setShapeInfo(T0, Shape0.t()); in distributeTransposes()
750 setShapeInfo(T1, Shape1.t()); in distributeTransposes()
832 setShapeInfo(Result, Shape0); in sinkTranspose()
853 setShapeInfo(Result, Shape0); in sinkTranspose()
888 setShapeInfo(M, {C, R}); in liftTranspose()
902 setShapeInfo(Add, {C, R}); in liftTranspose()