Home
last modified time | relevance | path

Searched defs:iterator_id (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_tf_data_stats.cc102 int64_t iterator_id = iterator_id_stat->IntValue(); in ProcessEventForest() local
124 int64_t iterator_id = iterator_id_stat->IntValue(); in ProcessEventForest() local
162 int64_t iterator_id = iterator_id_stat->IntValue(); in ProcessIteratorEvent() local
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/
H A D_hook_iterator.py70 def _check_iterator_valid(datapipe, iterator_id, next_method_exists=False) -> None: argument
144 def __init__(self, iterator, datapipe, iterator_id, has_next_method): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/
H A Dgroup_events.cc711 absl::optional<XStatVisitor> iterator_id = in ConnectTfDataEvents() local
747 absl::optional<XStatVisitor> iterator_id = in ConnectTfDataEvents() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py831 def _normalize_id(self, iterator_id): argument
848 def get_iterator(self, iterator_id): argument
857 def iterator_completed(self, iterator_id): argument
1031 def generator_py_func(iterator_id): argument
1097 def generator_py_func(iterator_id): argument
1127 def finalize_py_func(iterator_id): argument
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/
H A Dcombining.py378 def _check_valid_iterator_id(self, iterator_id) -> bool: argument