Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeDepthwise.cpp75 auto revisedWeightShapeType = RankedTensorType::get( in matchAndRewrite() local
80 op.getLoc(), revisedWeightShapeType, weight, in matchAndRewrite()
H A DTosaDecomposeConv2D.cpp70 auto revisedWeightShapeType = RankedTensorType::get( in matchAndRewrite() local
75 op.getLoc(), revisedWeightShapeType, weight, in matchAndRewrite()