func.func @gather_nd(%arg0: tensor<*x!tf_type.resource<tensor<80xf32>>>,
                     %arg1: tensor<3xf32>) {
  // CHECK: ResourceGatherNd
  // CHECK: tf_device.cluster
  // CHECK: Add
  // CHECK: ResourceGatherNd
  %0 = "tf.Const"() {value = dense<32> : tensor<i32>} : () -> tensor<i32>
  %1 = "tf.ResourceGatherNd"(%arg0, %0) {
    Tindices = i32
  } : (tensor<*x!tf_type.resource<tensor<80xf32>>>, tensor<i32>) -> tensor<1x80xf32>

This op gets the step id for each loop iteration.
  }];

  let arguments = (ins);

  let results = (outs
    TF_Int64Tensor:$iter_id
  );
}

def TF_ResourceGatherNdOp : TF_Op<"ResourceGatherNd", []> {
  let summary = "GatherNd on a resource.";

  let description = [{
This op reads the variable referenced by the first argument, and
then performs a GatherNd operation on it.
  }];