| /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)
303 [outputs, next_beam_state, beam_state, log_probs])
312 expected_log_probs.append(state_.log_probs[0][[1, 0, 0]])
313 expected_log_probs.append(state_.log_probs[1][[2, 1, 0]]) # 0 --> 1
320 self.assertAllEqual(next_state_.log_probs, expected_log_probs)
326 log_probs=nn_ops.log_softmax(
345 log_probs = nn_ops.log_softmax(logits)
360 [outputs, next_beam_state, beam_state, log_probs])
369 expected_log_probs.append(state_.log_probs[0][[1, 0, 0]])
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | mixture.py | 259 cat_probs = self._cat_probs(log_probs=False)
272 cat_probs = self._cat_probs(log_probs=False)
293 cat_log_probs = self._cat_probs(log_probs=True)
306 cat_log_probs = self._cat_probs(log_probs=True)
491 cat_probs = self._cat_probs(log_probs=False)
498 def _cat_probs(self, log_probs): argument
500 which_softmax = nn_ops.log_softmax if log_probs else nn_ops.softmax
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | hierarchical_controller.py | 578 log_probs = nn_ops.sparse_softmax_cross_entropy_with_logits(
580 log_probs = array_ops.reshape(log_probs, [batch_size, -1])
581 log_probs = math_ops.reduce_sum(log_probs, 1)
583 grouping_log_probs = log_probs
700 actions, log_probs = {}, {}
701 actions["sample"], log_probs["sample"] = (
704 actions["target"], log_probs["target"] = (
712 actions["greedy"], log_probs["greedy"] = (
721 return actions, log_probs
876 def body(i, prev_c, prev_h, actions, log_probs): argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 720 log_probs = array_ops.one_hot( # shape(batch_sz, beam_sz)
733 log_probs=log_probs,
895 log_probs = array_ops.one_hot( # shape(batch_sz, beam_sz)
908 log_probs=log_probs,
978 total_probs = array_ops.expand_dims(beam_state.log_probs, 2) + step_log_probs
1006 log_probs=total_probs,
1099 log_probs=next_beam_probs,
1157 def _get_scores(log_probs, sequence_lengths, length_penalty_weight, argument
1183 length_penalty_ = math_ops.cast(length_penalty_, dtype=log_probs.dtype)
1184 scores = log_probs / length_penalty_
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 130 std::vector<float>* log_probs, bool merge_repeated) const;
398 int n, std::vector<std::vector<int>>* paths, std::vector<float>* log_probs, in TopPaths() argument
401 TFLITE_DCHECK(log_probs); in TopPaths()
403 log_probs->clear(); in TopPaths()
423 log_probs->push_back(e->newp.total); in TopPaths()
|
| D | ctc_beam_search_decoder.cc | 207 std::vector<float> log_probs; in Eval() local
223 &log_probs, merge_repeated)); in Eval()
228 log_probabilities_output[b * top_paths + bp] = log_probs[bp]; in Eval()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 132 std::vector<float>* log_probs, bool merge_repeated) const;
401 int n, std::vector<std::vector<int>>* paths, std::vector<float>* log_probs, in TopPaths() argument
404 CHECK_NOTNULL(log_probs)->clear(); in TopPaths()
425 log_probs->push_back(e->newp.total); in TopPaths()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 308 std::vector<float> log_probs; in Compute() local
322 &log_probs, merge_repeated_)); in Compute()
327 log_prob_t(b, bp) = log_probs[bp]; in Compute()
|
| /external/tensorflow/tensorflow/contrib/gan/python/losses/python/ |
| D | losses_impl.py | 803 log_probs = [math_ops.reduce_mean(dist.log_prob(noise)) for dist, noise in
806 log_probs, weights, scope, loss_collection=loss_collection,
|