Searched refs:exp_logits (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_test.py | 560 exp_logits = np.concatenate([true_logits, sampled_logits], axis=1)
564 return weights, biases, hidden_acts, sampled_vals, exp_logits, exp_labels
599 (weights, biases, hidden_acts, sampled_vals, exp_logits,
623 self.assertEqual(exp_logits.shape, got_logits.shape, self._eps)
635 (weights, biases, hidden_acts, sampled_vals, exp_logits,
659 self.assertAllClose(exp_logits, got_logits, self._eps)
716 (weights, biases, hidden_acts, sampled_vals, exp_logits,
740 self.assertAllClose(exp_logits, got_logits, self._eps)
752 (weights, biases, hidden_acts, sampled_vals, exp_logits,
778 self.assertAllClose(exp_logits, got_logits, self._eps)
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | onehot_categorical_test.py | 166 exp_logits = np.exp(logits)
167 return exp_logits / exp_logits.sum(axis=-1, keepdims=True)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_xent_op.h | 100 typename TTypes<const T, 2>::Tensor32Bit exp_logits, in SparseXentGradGenerator() argument
104 : exp_logits_(exp_logits), in SparseXentGradGenerator()
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | categorical_test.py | 469 exp_logits = np.exp(logits)
470 return exp_logits / exp_logits.sum(axis=-1, keepdims=True)
|