Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Tosa/IR/
DTosaOps.cpp230 auto outputShapedType = outputType.dyn_cast<RankedTensorType>(); in buildMatMulOpWithQuantInfo() local
231 assert(outputShapedType && "Output must be a ranked tensor type"); in buildMatMulOpWithQuantInfo()
233 auto outputShape = outputShapedType.getShape(); in buildMatMulOpWithQuantInfo()
/external/llvm-project/mlir/lib/Dialect/Tosa/Utils/
DQuantUtils.cpp263 auto outputShapedType = outputType.dyn_cast<RankedTensorType>(); in buildConvOpResultTypeInfo() local
264 assert(outputShapedType && in buildConvOpResultTypeInfo()
267 auto outputShape = outputShapedType.getShape(); in buildConvOpResultTypeInfo()