Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dreshape_control_fusion.mlir28 // CHECK: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]]
32 // CHECK-SAME: ins(%[[RESHAPE]], %[[ARG1]] : tensor<?x?xf32>, tensor<?xf32>)
H A Ddrop-unit-extent-dims.mlir351 // CHECK-DAG: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]] {{\[}}[0, 1, 2], [3]]
357 // CHECK-SAME: ins(%[[RESHAPE]] : tensor<?x?xf32>)
384 // CHECK-DAG: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]] {{\[}}[0, 1, 2, 3]
390 // CHECK-SAME: ins(%[[RESHAPE]] : tensor<?xf32>)
419 // CHECK-DAG: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]] {{\[}}[0, 1], [2, 3]]
425 // CHECK-SAME: ins(%[[RESHAPE]] : tensor<?x?xf32>)
449 // CHECK: %[[RESHAPE:.+]] = tensor.collapse_shape %{{.+}} []
450 // CHECK: %[[RESULT:.+]] = tensor.insert_slice %[[RESHAPE]]
H A Dfuse-with-reshape-by-collapsing.mlir271 // CHECK: %[[RESHAPE:.+]] = tensor.expand_shape %[[ARG0]]
273 // CHECK-SAME: ins(%[[RESHAPE]], %[[ARG1]] :
298 // CHECK: %[[RESHAPE:.+]] = tensor.expand_shape %[[ARG0]]
300 // CHECK-SAME: ins(%[[RESHAPE]], %[[ARG1]] :
325 // CHECK: %[[RESHAPE:.+]] = tensor.expand_shape %[[ARG0]]
327 // CHECK-SAME: ins(%[[RESHAPE]], %[[ARG1]] :
H A Dreshape_fusion.mlir471 // CHECK: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]]
473 // CHECK-SAME: ins(%[[RESHAPE]] : tensor<?xf32>)
498 // CHECK: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]]
500 // CHECK-SAME: ins(%[[RESHAPE]], %[[ARG1]] : tensor<2xi64>, tensor<?xi64>)
H A Dcanonicalize.mlir412 // CHECK: %[[RESHAPE:.+]] = tensor.collapse_shape %[[ARG0]]
414 // CHECK: %[[RESULT:.+]] = linalg.fill ins(%{{.+}}{{.*}}outs(%[[RESHAPE]]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Transforms/
H A DPasses.td46 number of dimensions. Insert RESHAPE operations to prepend dimensions
/llvm-project-15.0.7/mlir/test/Conversion/TosaToLinalg/
H A Dtosa-to-linalg.mlir92 // CHECK: [[RESHAPE:%.+]] = tensor.collapse_shape %arg0
93 …[[$MAP0]], #[[$MAP1]], #[[$MAP1]]], iterator_types = ["parallel"]} ins([[RESHAPE]], %arg1 : tensor…
110 // CHECK: [[RESHAPE:%.+]] = tensor.collapse_shape %arg1
547 // CHECK: [[RESHAPE:%.+]] = tensor.collapse_shape %arg0 {{\[}}[0, 1]]
549 // CHECK: return [[RESHAPE]]
557 // CHECK: [[RESHAPE:%.+]] = tensor.collapse_shape %arg0 {{\[}}[0, 1]]
559 // CHECK: return [[RESHAPE]]
567 // CHECK: [[RESHAPE:%.+]] = tensor.expand_shape %arg0 {{\[}}[0, 1]]
569 // CHECK: return [[RESHAPE]]
577 // CHECK: [[RESHAPE:%.+]] = tensor.expand_shape %arg0 {{\[}}[0, 1]]
[all …]
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dfold-implementation.h68 Expr<T> RESHAPE(FunctionRef<T> &&);
848 template <typename T> Expr<T> Folder<T>::RESHAPE(FunctionRef<T> &&funcRef) { in RESHAPE() function
1130 return Folder<T>{context}.RESHAPE(std::move(funcRef)); in FoldOperation()