Move tests for arg and retval devices from arg-retval-attrs.pbtxt to existing device attribute test (NFC).

PiperOrigin-RevId: 346111552
Change-Id: Iaac59689e23b64416d1a0836f5c429ce52030ab4

Author: andyly

tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/arg-retval-attrs.pbtxt

@@ -19,7 +19,6 @@ node {
 node {
   name: "arg1"
   op: "_Arg"
-  device: "CPU:1"
   attr {
     key: "T"
     value {
@@ -111,7 +110,6 @@ node {
   name: "ret2"
   op: "_Retval"
   input: "arg2"
-  device: "CPU:0"
   attr {
     key: "T"
     value {
@@ -152,6 +150,6 @@ versions {
 # arg/result attributes, at the right index.
 
 # CHECK:      func @main
-# CHECK-SAME: ({{%.*}}: tensor<*xf32>, {{%.*}}: tensor<*xi32> {tf._arg1_attr0 = "_arg1_attr0_value", tf._arg1_attr1 = 8.000000e+00 : f32, tf.device = "CPU:1"}, {{%.*}}: tensor<*xi1>)
-# CHECK-SAME: -> (tensor<*xf32> {tf._ret0_attr0 = 8 : i64, tf._ret0_attr1 = false}, tensor<*xi32>, tensor<*xi1> {tf._ret2_attr0 = !tf.variant, tf._ret2_attr1 = #tf.shape<128x1024>, tf.device = "CPU:0"})
+# CHECK-SAME: ({{%.*}}: tensor<*xf32>, {{%.*}}: tensor<*xi32> {tf._arg1_attr0 = "_arg1_attr0_value", tf._arg1_attr1 = 8.000000e+00 : f32}, {{%.*}}: tensor<*xi1>)
+# CHECK-SAME: -> (tensor<*xf32> {tf._ret0_attr0 = 8 : i64, tf._ret0_attr1 = false}, tensor<*xi32>, tensor<*xi1> {tf._ret2_attr0 = !tf.variant, tf._ret2_attr1 = #tf.shape<128x1024>})
 # CHECK-SAME: attributes {tf.entry_function = {control_outputs = "", inputs = "arg0,arg1,arg2", outputs = "ret0,ret1,ret2"}}

tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/device-arg-attr.pbtxt renamed to tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/device-arg-retval-attr.pbtxt

@@ -1,9 +1,9 @@
 # RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function -o - | FileCheck %s
 
-# Verify arg devices are added as arg attributes.
+# Verify arg and ret devices are added as arg and ret attributes.
 
 # CHECK-LABEL: func @main
-# CHECK-SAME:  (%[[ARG_0:[a-z0-9]+]]: tensor<*xf32> {tf.device = "/CPU:0"}, %[[ARG_1:[a-z0-9]+]]: tensor<2x4x6x8xi32>) -> (tensor<*xf32>, tensor<*xi32>)
+# CHECK-SAME:  (%[[ARG_0:[a-z0-9]+]]: tensor<*xf32> {tf.device = "/CPU:0"}, %[[ARG_1:[a-z0-9]+]]: tensor<2x4x6x8xi32>) -> (tensor<*xf32>, tensor<*xi32> {tf.device = "/CPU:1"})
 
 node {
   name: "args_0"
@@ -95,6 +95,7 @@ node {
   name: "rets_1"
   op: "_Retval"
   input: "identity:1"
+  device: "/CPU:1"
   attr {
     key: "T"
     value {