Add generated import of tf.DeviceIndex.

PiperOrigin-RevId: 316178064
Change-Id: I4fa2186c74b091a6b901fe5c3af8d210193daf4d

tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

@@ -2324,6 +2324,18 @@ horizontal and vertices strides, `strides = [1, stride, stride, 1]`.
   TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
 }
 
+def TF_DeviceIndexOp : TF_Op<"DeviceIndex", [NoSideEffect]> {
+  let summary = "Return the index of device the op runs.";
+
+  let arguments = (ins
+    StrArrayAttr:$device_names
+  );
+
+  let results = (outs
+    I32Tensor:$index
+  );
+}
+
 def TF_DiagPartOp : TF_Op<"DiagPart", [NoSideEffect]> {
   let summary = "Returns the diagonal part of the tensor.";