Home
last modified time | relevance | path

Searched refs:fullyConnectedShape (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp80 llvm::SmallVector<int64_t, 2> fullyConnectedShape{combined, weightShape[0]}; in matchAndRewrite() local
82 RankedTensorType::get(fullyConnectedShape, resultType.getElementType()); in matchAndRewrite()