search

wala / ML

Eclipse Public License 2.0
23 stars 17 forks source link

Track tensors emanating from `tf.data.Dataset.enumerate()` #165

Open khatchad opened 3 months ago

khatchad commented 3 months ago

Since https://github.com/wala/ML/pull/132, we can enumerate() over datasets and track the tensors. However, the TF Data API has a method tf.data.Dataset.enumerate() that does is similar to Python's enumerate(). But, we don't track tensors coming out of that API yet.

Related to #164.

khatchad commented 3 months ago

zip() looks very similar.

khatchad commented 3 months ago

Seems to work when using Jython 3.