// 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")

    inst->removeAttr(attr_id);
  } else {
    // Some control flow ops in TensorFlow Graph have their respective "Ref" ops
    // as well. For example there is Enter and RefEnter op. RefEnter forwards
    // the input ref buffer to output. However both Enter and RefEnter are
    // mapped to tf_executor::EnterOp during import. Check if it is a Ref op to
    // correctly map to the TensorFlow Graph op.