Exemplos de enqueue_tpu_embedding_sparse_batch em Python

Exemplo n.º 1

Arquivo: tpu_embedding.py Projeto: meharm1990/tensorflowsample

 def _generate_enqueue_op(self, sparse_features, device_ordinal):
     with ops.colocate_with(list(sparse_features.values())[0]):
         sample_idcs, embedding_idcs, aggregation_weights = (
             self._format_for_tpu_embedding_sparse_batch(sparse_features))
         return tpu_ops.enqueue_tpu_embedding_sparse_batch(
             sample_idcs,
             embedding_idcs,
             aggregation_weights,
             combiners=self._combiners,
             device_ordinal=device_ordinal)

Exemplo n.º 2

Arquivo: tpu_embedding.py Projeto: terrytangyuan/tensorflow

 def _generate_enqueue_op(self, sparse_features, device_ordinal):
   with ops.colocate_with(list(sparse_features.values())[0]):
     sample_idcs, embedding_idcs, aggregation_weights = (
         self._format_for_tpu_embedding_sparse_batch(sparse_features))
     return tpu_ops.enqueue_tpu_embedding_sparse_batch(
         sample_idcs,
         embedding_idcs,
         aggregation_weights,
         combiners=self._combiners,
         device_ordinal=device_ordinal)