Lines Matching refs:setShapeInfo
591 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo() function in __anon821fcdb70111::LowerMatrixIntrinsics
657 Propagate = setShapeInfo(Inst, *SI); in propagateShapeForward()
701 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
704 if (setShapeInfo(MatrixB, {N, K})) in propagateShapeBackward()
710 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
715 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
728 if (setShapeInfo(U.get(), Shape)) in propagateShapeBackward()
755 setShapeInfo(T0, Shape0.t()); in distributeTransposes()
758 setShapeInfo(T1, Shape1.t()); in distributeTransposes()
840 setShapeInfo(Result, Shape0); in sinkTranspose()
861 setShapeInfo(Result, Shape0); in sinkTranspose()
896 setShapeInfo(M, {C, R}); in liftTranspose()
912 setShapeInfo(Add, {R, C}); in liftTranspose()