forked from OSchip/llvm-project
11 lines
435 B
MLIR
11 lines
435 B
MLIR
// RUN: mlir-opt <%s | mlir-opt | FileCheck %s
|
|
|
|
// CHECK-LABEL: func @extract(
|
|
// CHECK-SAME: %[[TENSOR:.*]]: tensor<?x?x?xf32>,
|
|
// CHECK-SAME: %[[INDEX:.*]]: index) {
|
|
func @extract(%arg0: tensor<?x?x?xf32>, %arg1: index) {
|
|
// CHECK: tensor.extract %[[TENSOR]][%[[INDEX]], %[[INDEX]], %[[INDEX]]] : tensor<?x?x?xf32>
|
|
%0 = tensor.extract %arg0[%arg1, %arg1, %arg1] : tensor<?x?x?xf32>
|
|
return
|
|
}
|