Lines Matching defs:ShapeInfo

197 struct ShapeInfo {
203   ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0)
207   ShapeInfo(Value *NumRows, Value *NumColumns)
208       : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(),
211   bool operator==(const ShapeInfo &other) {
214   bool operator!=(const ShapeInfo &other) { return !(*this == other); }
236   ShapeInfo t() const { return ShapeInfo(NumColumns, NumRows); }
259 /// Return the ShapeInfo for the result of \p I, it it can be determined.
260 static std::optional<ShapeInfo>
262                         const ValueMap<Value *, ShapeInfo> &ShapeMap) {
268     return ShapeInfo(M, K);
272     return ShapeInfo(N, M);
277     return ShapeInfo(N, M);
280     return ShapeInfo(M, N);
498   ValueMap<Value *, ShapeInfo> ShapeMap;
552   MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI,
591   bool setShapeInfo(Value *V, ShapeInfo Shape) {
726         ShapeInfo Shape = ShapeMap[V];
747       Value *Op0, ShapeInfo Shape0, Value *Op1, ShapeInfo Shape1,
749       function_ref<Instruction *(Value *, ShapeInfo, Value *, ShapeInfo)>
813           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
835           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
855           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
1125                       bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) {
1148                       ShapeInfo MatrixShape, Value *I, Value *J,
1149                       ShapeInfo ResultShape, Type *EltTy,
1166                  bool IsVolatile, ShapeInfo Shape) {
1190                    MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape,
1228                   Value *Stride, bool IsVolatile, ShapeInfo Shape) {
1338     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1339     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1696     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1697     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1736   void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape,
1737                         Value *RPtr, ShapeInfo RShape, StoreInst *Store) {
1803     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1804     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1882       ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1883       ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1893         MA = getMatrix(A, ShapeInfo(R, M), Builder);
1894         MB = getMatrix(T, ShapeInfo(C, M), Builder);
1897         MA = getMatrix(T, ShapeInfo(R, M), Builder);
1898         MB = getMatrix(B, ShapeInfo(C, M), Builder);
2013     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
2014     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
2041     ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2));
2107     ShapeInfo &Shape = I->second;
2157     ShapeInfo &Shape = I->second;