Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp351 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
363 inferredReturnShapes.push_back(ShapedTypeComponents(outShape)); in inferReturnTypeComponents()
399 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
494 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
502 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
561 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
565 inferredReturnShapes.resize(1); in inferReturnTypeComponents()
566 inputShape.getDims(inferredReturnShapes[0]); in inferReturnTypeComponents()
654 inferredReturnShapes.push_back(ShapedTypeComponents()); in inferReturnTypeComponents()
842 inferredReturnShapes.push_back(ShapedTypeComponents()); in ReduceInferReturnTypes()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.td152 $inferredReturnShapes),
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp1164 SmallVectorImpl<ShapedTypeComponents> &inferredReturnShapes) { in inferReturnTypeComponents() argument
1174 inferredReturnShapes.push_back(ShapedTypeComponents({dim}, type)); in inferReturnTypeComponents()