Home
last modified time | relevance | path

Searched refs:enqueue_datas (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py225 enqueue_datas = collections.OrderedDict(
228 enqueue_datas_list.append(enqueue_datas)
961 enqueue_datas,
964 ) for i, enqueue_datas in enumerate(enqueue_datas_list)
972 for i, enqueue_datas in enumerate(enqueue_datas_list):
973 used_feature_set = set(enqueue_datas.keys())
990 for feature, enqueue_data in six.iteritems(enqueue_datas):
1040 self, enqueue_datas, device_ordinal, mode_override=None): argument
1041 enqueue_data0 = list(enqueue_datas.values())[0]
1047 **self._format_for_tpu_embedding_sparse_tensor_batch(enqueue_datas)
[all …]