Searched refs:log_softmax (Results 1 – 25 of 25) sorted by relevance
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | nn.py | 36 log_softmax = core.define_unary_op('log_softmax', nn.log_softmax) variable
|
| D | nn_test.py | 44 ('log_softmax', nn_ops.log_softmax, nn.log_softmax),
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | relaxed_onehot_categorical.py | 267 samples = nn_ops.log_softmax(noisy_logits)
289 log_softmax = nn_ops.log_softmax(logits_2d - x_2d * self._temperature_2d)
290 log_unnorm_prob = math_ops.reduce_sum(log_softmax, [-1], keepdims=False)
|
| D | onehot_categorical.py | 217 nn_ops.log_softmax(self.logits) * self.probs, axis=-1)
269 nn_ops.softmax(a.logits) * (nn_ops.log_softmax(a.logits)
270 - nn_ops.log_softmax(b.logits)),
|
| D | mixture_same_family.py | 266 log_mix_prob = nn_ops.log_softmax(
283 log_mix_prob = nn_ops.log_softmax(
|
| D | mixture.py | 500 which_softmax = nn_ops.log_softmax if log_probs else nn_ops.softmax
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | categorical.py | 319 nn_ops.log_softmax(self.logits) * self.probs, axis=-1)
344 delta_log_probs1 = (nn_ops.log_softmax(a.logits) -
345 nn_ops.log_softmax(b.logits))
|
| D | multinomial.py | 285 return math_ops.reduce_sum(counts * nn_ops.log_softmax(self.logits), -1)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 269 log_probs=nn_ops.log_softmax(
288 log_probs = nn_ops.log_softmax(logits)
326 log_probs=nn_ops.log_softmax(
345 log_probs = nn_ops.log_softmax(logits)
446 log_probs = nn_ops.log_softmax(logits)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | softmax_op_test.py | 65 tf_softmax = nn_ops.log_softmax(np_features, axis=dim, name=name)
116 tf_log_softmax = nn_ops.log_softmax(features)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | BUILD | 323 "reference/integer_ops/log_softmax.h",
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| D | classifier_metrics_impl.py | 187 p * (nn_ops.log_softmax(p_logits) - math_ops.log(q)), axis=1)
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_compatibility.md | 58 * [tf.nn.log_softmax](https://www.tensorflow.org/api_docs/python/tf/nn/log_softmax) -
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | categorical_test.py | 344 log_probabilities = nn_ops.log_softmax(logits)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2932 def log_softmax(logits, axis=None, name=None, dim=None): function
2957 return _softmax(logits, gen_nn_ops.log_softmax, axis, name)
2984 return _softmax(logits, gen_nn_ops.log_softmax, axis, name)
|
| D | ctc_ops.py | 553 ilabel_log_probs = nn_ops.log_softmax(logits)
|
| D | nn_grad.py | 542 return grad, _BroadcastMul(grad_loss, -nn_ops.log_softmax(logits))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 200 name: "log_softmax"
|
| D | tensorflow.math.pbtxt | 236 name: "log_softmax"
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 271 "log_softmax",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 236 name: "log_softmax"
|
| D | tensorflow.nn.pbtxt | 244 name: "log_softmax"
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 976 step_log_probs = nn_ops.log_softmax(logits)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 330 nn_ops.log_softmax,
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 1197 out = tf.nn.log_softmax(input_tensor)
|