Improve performance (#2529)

DLPerf · web-flow · commit ddfa3b7fd917 · 2021-09-03T09:41:15.000+08:00
diff --git a/elasticdl/python/tests/data_reader_test.py b/elasticdl/python/tests/data_reader_test.py
@@ -63,12 +63,13 @@ def test_recordio_data_reader(self):
                 )
             )
             self.assertEqual(len(records), num_records)
+            fixed_len_feature = tf.io.FixedLenFeature([1], tf.float32)
             for record in records:
                 parsed_record = tf.io.parse_single_example(
                     record,
                     {
-                        "x": tf.io.FixedLenFeature([1], tf.float32),
-                        "y": tf.io.FixedLenFeature([1], tf.float32),
+                        "x": fixed_len_feature,
+                        "y": fixed_len_feature,
                     },
                 )
                 for k, v in parsed_record.items():

Original file line number	Diff line number	Diff line change
`@@ -63,12 +63,13 @@ def test_recordio_data_reader(self):`
`63`	`63`	`)`
`64`	`64`	`)`
`65`	`65`	`self.assertEqual(len(records), num_records)`
	`66`	`+ fixed_len_feature = tf.io.FixedLenFeature([1], tf.float32)`
`66`	`67`	`for record in records:`
`67`	`68`	`parsed_record = tf.io.parse_single_example(`
`68`	`69`	`record,`
`69`	`70`	`{`
`70`		`- "x": tf.io.FixedLenFeature([1], tf.float32),`
`71`		`- "y": tf.io.FixedLenFeature([1], tf.float32),`
	`71`	`+ "x": fixed_len_feature,`
	`72`	`+ "y": fixed_len_feature,`
`72`	`73`	`},`
`73`	`74`	`)`
`74`	`75`	`for k, v in parsed_record.items():`