Extract the lambda declaration to avoid the autograph warning. A lambda might...

Extract the lambda declaration to avoid the autograph warning. A lambda might not be at all necessary, so trying that fist.

PiperOrigin-RevId: 309353494

official/vision/detection/dataloader/input_reader.py

@@ -91,7 +91,7 @@ class InputFn(object):
       dataset = dataset.repeat()
 
     dataset = dataset.interleave(
-        map_func=lambda file_name: self._dataset_fn(file_name), cycle_length=32,
+        map_func=self._dataset_fn, cycle_length=32,
         num_parallel_calls=tf.data.experimental.AUTOTUNE)
 
     if self._is_training: