Searched defs:resultShape (Results 1 – 11 of 11) sorted by relevance
235 Value resultShape = b.create<tensor::FromElementsOp>(loc, resultDims); in runOnOperation() local
57 const ArrayRef<int64_t> resultShape = resultTy.getShape(); in matchAndRewrite() local
156 Optional<Value> resultShape; in tryLowerTo1DOr2DReduction() local
362 SmallVector<Value, 1> resultShape; in materializeFinalReshape() local
2867 auto resultShape = resultType.getShape(); in matchAndRewrite() local
99 auto resultShape = resultType.getShape(); in matchAndRewrite() local
508 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
1286 SmallVector<int64_t> resultShape = to_vector(operandShape); in inferReturnTypeComponents() local2773 SmallVector<int64_t> resultShape(operandRankedType.getShape().begin(), in inferReturnTypeComponents() local5734 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local7411 SmallVector<int64_t> resultShape; in inferReturnTypes() local
481 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
1249 SmallVector<int64_t> resultShape = to_vector(operandShape); in inferReturnTypeComponents() local2256 SmallVector<int64_t> resultShape(operandRankedType.getShape().begin(), in inferReturnTypeComponents() local4328 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local4977 SmallVector<int64_t> resultShape; in inferReturnTypes() local
1604 SmallVector<int64_t> resultShape = llvm::to_vector(llvm::map_range( in inferReturnTypes() local1713 const SmallVector<int64_t> resultShape = llvm::to_vector(llvm::map_range( in inferReturnTypes() local