Searched refs:enqueue_data (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding.py | 1759 def _check_agreement(data, name, feature, enqueue_data): argument
1762 data.device != enqueue_data.embedding_indices.device):
1787 for feature, enqueue_data in enqueue_datas.items():
1791 if isinstance(enqueue_data, EnqueueData):
1792 if enqueue_data.sample_indices is None and combiner:
1797 _check_agreement(enqueue_data.sample_indices, 'sample_indices',
1798 feature, enqueue_data)
1799 _check_agreement(enqueue_data.aggregation_weights,
1800 'aggregation_weights', feature, enqueue_data)
1802 elif isinstance(enqueue_data, RaggedEnqueueData):
[all …]
|