Searched refs:vals_per_batch_entry (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 674 vals_per_batch_entry = np.random.randint(
676 num_vals = np.sum(vals_per_batch_entry)
682 for batch_entry, num_val in enumerate(vals_per_batch_entry):
697 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
699 def _GroupByBatchEntry(self, vals, vals_per_batch_entry): argument
702 for num_val in vals_per_batch_entry:
713 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
716 grouped_ids = self._GroupByBatchEntry(ids, vals_per_batch_entry)
717 grouped_weights = self._GroupByBatchEntry(weights, vals_per_batch_entry)
719 np.ones(np.sum(vals_per_batch_entry)), vals_per_batch_entry)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 652 vals_per_batch_entry = np.random.randint(
654 num_vals = np.sum(vals_per_batch_entry)
660 for batch_entry, num_val in enumerate(vals_per_batch_entry):
675 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
677 def _GroupByBatchEntry(self, vals, vals_per_batch_entry): argument
680 for num_val in vals_per_batch_entry:
692 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
695 grouped_ids = self._GroupByBatchEntry(ids, vals_per_batch_entry)
696 grouped_weights = self._GroupByBatchEntry(weights, vals_per_batch_entry)
698 np.ones(np.sum(vals_per_batch_entry)), vals_per_batch_entry)
|