Lines Matching refs:ShapeInfo

344   struct ShapeInfo {  struct in __anon0f2dcaa70111::LowerMatrixIntrinsics
350 ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0) in ShapeInfo() function
354 ShapeInfo(Value *NumRows, Value *NumColumns) in ShapeInfo() argument
355 : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(), in ShapeInfo()
358 bool operator==(const ShapeInfo &other) { in operator ==() argument
361 bool operator!=(const ShapeInfo &other) { return !(*this == other); } in operator !=() argument
389 DenseMap<Value *, ShapeInfo> ShapeMap;
425 MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI, in getMatrix()
465 bool setShapeInfo(Value *V, ShapeInfo Shape) { in setShapeInfo()
646 ShapeInfo Shape = ShapeMap[V]; in propagateShapeBackward()
800 bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) { in loadMatrix()
821 ShapeInfo MatrixShape, Value *I, Value *J, in loadMatrix()
822 ShapeInfo ResultShape, Type *EltTy, in loadMatrix()
845 bool IsVolatile, ShapeInfo Shape) { in LowerLoad()
869 MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape, in storeMatrix()
911 Value *Stride, bool IsVolatile, ShapeInfo Shape) { in LowerStore()
1174 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in isFusionProfitable()
1175 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in isFusionProfitable()
1212 void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape, in createTiledLoops()
1213 Value *RPtr, ShapeInfo RShape, StoreInst *Store, in createTiledLoops()
1277 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in emitSIMDTiling()
1278 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in emitSIMDTiling()
1370 ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3)); in LowerMultiply()
1371 ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4)); in LowerMultiply()
1399 ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2)); in LowerTranspose()
1464 ShapeInfo &Shape = I->second; in VisitBinaryOperator()