Home
last modified time | relevance | path

Searched defs:inputShape (Results 1 – 14 of 14) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp346 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
428 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
487 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
558 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
576 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
609 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
647 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
746 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
826 ShapeAdaptor inputShape = operands.getShape(2); in inferReturnTypeComponents() local
974 ShapeAdaptor inputShape = operands.getShape(0); in NARY_SHAPE_INFER() local
[all …]
H A DTosaCanonicalizations.cpp105 ShapedType inputShape = input.getType().cast<ShapedType>(); in matchAndRewrite() local
316 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaFoldConstantTranspose.cpp49 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
H A DTosaDecomposeConv2D.cpp54 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
H A DTosaDecomposeDepthwise.cpp60 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
/llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/
H A DDialect.cpp294 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/
H A DDialect.cpp294 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/
H A DDialect.cpp412 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/
H A DDialect.cpp412 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/
H A DDialect.cpp412 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/
H A DDialect.cpp467 auto inputShape = inputType.getShape(); in verify() local
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp41 auto inputShape = inputTy.getShape(); in applyPad() local
H A DTosaToLinalg.cpp1780 auto inputShape = inputTy.getShape(); in matchAndRewrite() local
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp96 ArrayRef<int64_t> inputShape = inputType.getShape(); in handleMultidimensionalVectors() local