From 778548552bff1dcc1697c5b7f47330ccd4599621 Mon Sep 17 00:00:00 2001
From: DLPerf <88604684+DLPerf@users.noreply.github.com>
Date: Thu, 26 Aug 2021 17:41:27 +0800
Subject: [PATCH] Update input_utils.py

---
 tensorrec/input_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorrec/input_utils.py b/tensorrec/input_utils.py
index 405a648..021a836 100644
--- a/tensorrec/input_utils.py
+++ b/tensorrec/input_utils.py
@@ -123,5 +123,5 @@ def parse_tensorrec_tfrecord(example_proto):
         return (parsed_features['row_index'], parsed_features['col_index'], parsed_features['values'],
                 parsed_features['d0'], parsed_features['d1'])
 
-    dataset = tf.data.TFRecordDataset(tfrecord_path).map(parse_tensorrec_tfrecord)
+    dataset = tf.data.TFRecordDataset(tfrecord_path).map(parse_tensorrec_tfrecord,num_parallel_calls=tf.data.experimental.AUTOTUNE)
     return dataset