Lines Matching refs:setShapeInfo
489 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo() function in __anonf095cceb0111::LowerMatrixIntrinsics
573 Propagate = setShapeInfo(Inst, {M, K}); in propagateShapeForward()
577 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
581 Propagate = setShapeInfo(Inst, {N, M}); in propagateShapeForward()
585 Propagate = setShapeInfo(Inst, {M, N}); in propagateShapeForward()
589 setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
596 Propagate |= setShapeInfo(Inst, OpShape->second); in propagateShapeForward()
644 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
647 if (setShapeInfo(MatrixB, {N, K})) in propagateShapeBackward()
653 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
658 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
671 if (setShapeInfo(U.get(), Shape)) in propagateShapeBackward()
748 setShapeInfo(T0, {C, K}); in optimizeTransposes()
752 setShapeInfo(T1, {K, R}); in optimizeTransposes()
787 setShapeInfo(M, {C, R}); in optimizeTransposes()