// RUN: tf-mlir-translate -mlir-to-graphdef %s -o - | FileCheck %s

// Verify the ops generated when Ref type is used in a while loop.
func.func @main() {
  // CHECK:  op: "RefEnter"
  // CHECK:  op: "RefMerge"
  // CHECK:  op: "RefSwitch"
  // CHECK:  op: "RefExit"
  // CHECK:  op: "RefNextIteration"
  tf_executor.graph {
    %0:3 = tf_executor.NextIteration.Source : tensor<*x!tf_type.int32ref> {device = "", T = "tfdtype$DT_INT32"} loc("while/NextIteration")

 IsVariableInitialized 1
 IteratorGetNext 1
 IteratorV2 1
 Less 9
 LogicalAnd 3
 LoopCond 8
 Max 4
 Maximum 44
 Merge 145
 Minimum 43
 Mul 8
 NextIteration 136
 RandomUniform 14
 Range 9
 RefSwitch 166
 Reshape 2
 ScatterAdd 4
 Shape 6
 ShapeN 10
 Size 2
 Snapshot 1
 StackPopV2 314
 StackPushV2 314
 StackV2 314
 StridedSlice 12
 Sub 5
 Sum 2
 Switch 247
 TensorArrayGatherV3 8