Searched refs:enqueue_data (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1662 def _check_agreement(data, name, feature, enqueue_data): argument
1665 data.device != enqueue_data.embedding_indices.device):
1690 for feature, enqueue_data in six.iteritems(enqueue_datas):
1694 if isinstance(enqueue_data, EnqueueData):
1695 if enqueue_data.sample_indices is None and combiner:
1700 _check_agreement(enqueue_data.sample_indices, 'sample_indices',
1701 feature, enqueue_data)
1702 _check_agreement(enqueue_data.aggregation_weights,
1703 'aggregation_weights', feature, enqueue_data)
1705 elif isinstance(enqueue_data, RaggedEnqueueData):
[all …]
|