xref: /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/canonicalize/tuple.mlir (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1// RUN: mlir-hlo-opt %s -split-input-file -pass-pipeline='func.func(canonicalize)' | FileCheck %s
2
3// CHECK-LABEL: func @fold_access
4// CHECK-SAME: [[ARG:%[a-zA-Z0-9]+]]
5func.func @fold_access(%arg : tensor<i32>) -> tensor<i32> {
6  // CHECK-NEXT: return [[ARG]]
7  %tuple = "mhlo.tuple"(%arg) : (tensor<i32>) -> tuple<tensor<i32>>
8  %element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i32>>) -> tensor<i32>
9  func.return %element : tensor<i32>
10}
11