}
    }
  }
}
node {
  name: "ParseExample/ParseExampleV2"
  op: "ParseExampleV2"
  input: "input0"
  input: "ParseExample/ParseExampleV2/names"
  input: "ParseExample/ParseExampleV2/sparse_keys"
  input: "ParseExample/ParseExampleV2/dense_keys"
  input: "ParseExample/ParseExampleV2/ragged_keys"
  input: "ParseExample/Const"
  input: "ParseExample/Const_1"
  attr {
    key: "Tdense"
    value {

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

// CHECK:      name: "tf.ParseExample"
// CHECK-NEXT: op: "ParseExample"
// CHECK-NEXT: input: "tf.Const{{_.*_3}}"
// CHECK-NEXT: input: "tf.Const"
// CHECK-NEXT: input: "tf.Const{{_.*_1}}"
// CHECK-NEXT: input: "tf.Const{{_.*_2}}"
// CHECK-NEXT: attr {
// CHECK-NEXT:   key: "Ndense"
// CHECK-NEXT:   value {
// CHECK-NEXT:     i: 1
// CHECK-NEXT:   }
// CHECK-NEXT: }

      // CHECK:      name: "ParseExample"
      // CHECK-NEXT: op: "ParseExampleV2"
      // CHECK-NEXT: input: "input0"
      // CHECK-NEXT: input: "tf.Const{{_.*_3}}"
      // CHECK-NEXT: input: "tf.Const{{_.*_5}}"

        }
      }
    }
    node {
      name: "ParseExample/ParseExample"
      op: "ParseExample"
      input: "tf_example"
      input: "ParseExample/ParseExample/names"
      input: "ParseExample/ParseExample/dense_keys_0"
      input: "ParseExample/ParseExample/dense_keys_1"
      input: "ParseExample/Const"
      input: "ParseExample/Reshape"
      attr {
        key: "Ndense"
        value {

# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false -tf-output-arrays=result %s | FileCheck %s

# CHECK: %[[output:.*]], %[[control:.*]]tf_executor.island wraps "tf.ParseExample"
# CHECK: operandSegmentSizes = array<i32: 1, 1, 0, 1, 1>
# CHECK: resultSegmentSizes = array<i32: 0, 0, 0, 1>
# CHECK: tf_executor.fetch %[[output]] : tensor<*xi64>

node {
  name: "serilaized"
  op: "Const"
  attr {
    key: "dtype"

  ) -> (tensor<1x1x!tf_type.string>, tensor<1x1x!tf_type.string>)
  func.return %4#0, %4#1 : tensor<1x1x!tf_type.string>, tensor<1x1x!tf_type.string>
// CHECK: "tf.ParseExample"(
// CHECK-SAME: operandSegmentSizes = array<i32: 1, 1, 0, 2, 2>, resultSegmentSizes = array<i32: 0, 0, 0, 2>
}

// CHECK-LABEL: TfMapDataset

// CHECK-SAME: {{.*}} !tfrt.chain
// CHECK-SAME: [[serialized:%.*]]: !tfrt_fallback.tf_tensor
func.func @main(%serialized: tensor<32x!tf_type.string>) -> (tensor<?x2xi64>) attributes {tf.entry_function = {inputs = "input0", outputs = "ParseExample/ParseExampleV2"}} {
  %dense_default_0 = "tf.Const"() {device = "/device:CPU:0", dtype = f32, value = dense<[]> : tensor<0xf32>} : () -> tensor<0xf32>

    // expected-remark@+1 {{Cost: 1}}
    %11 = "tf.Const"() {value = dense<-1> : tensor<i32>} : () -> tensor<i32>
    // expected-remark@+1 {{Cost: 19}}

    }
    // TODO(b/204997177): Deprecate and remove.
    func::FuncOp func() {  return ResolveFunc(nullptr); }
  }];
}

def TF_ParseExampleOp : TF_Op<"ParseExample",
                               [Pure,
                                AttrSizedResultSegments,
                                AttrSizedOperandSegments]> {

  let summary =