Home
last modified time | relevance | path

Searched refs:reduce_logsumexp (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py482 label_olabels = math_ops.reduce_logsumexp(label_states + one_hot, axis=2)
483 blank_olabels = math_ops.reduce_logsumexp(
520 blank_olabels = math_ops.reduce_logsumexp(
939 return math_ops.reduce_logsumexp(states + one_hot, axis=-1)
978 state_log_prob = math_ops.reduce_logsumexp(state_log_prob, axis=-1)
980 log_prob_sum = math_ops.reduce_logsumexp(
995 state_log_prob = math_ops.reduce_logsumexp(state_log_prob, axis=-1)
997 log_prob_sum = math_ops.reduce_logsumexp(
1017 fwd_bwd_log_probs_sum = math_ops.reduce_logsumexp(
Dmath_ops_test.py101 y_tf_np = math_ops.reduce_logsumexp(x_np).eval()
109 y_tf = math_ops.reduce_logsumexp(x_np, axis=[0])
119 y_tf = math_ops.reduce_logsumexp(x_np, axis=0)
130 y_tf_np = math_ops.reduce_logsumexp(x_np, keepdims=True).eval()
149 y_tf_np = math_ops.reduce_logsumexp(x_tf).eval()
167 y_tf_np = math_ops.reduce_logsumexp(x_tf).eval()
174 res = math_ops.reduce_logsumexp(-np.inf).eval()
Dmath_ops.py2296 return reduce_logsumexp(input_tensor, axis, keepdims, name)
2300 def reduce_logsumexp(input_tensor, axis=None, keepdims=False, name=None): function
/external/tensorflow/tensorflow/contrib/crf/python/ops/
Dcrf.py149 return math_ops.reduce_logsumexp(
190 log_norm = math_ops.reduce_logsumexp(first_input, [1])
214 log_norm = math_ops.reduce_logsumexp(alphas, [1])
382 new_alphas = inputs + math_ops.reduce_logsumexp(transition_scores, [1])
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py90 self.assertAllClose(tf.reduce_logsumexp(a, reduction_indices=[1]).eval(),
93 tf.reduce_logsumexp(a, reduction_indices=[0, 1]).eval(),
96 tf.reduce_logsumexp(a, [0, 1]).eval(), 6.45619344711)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dpoisson_lognormal.py402 return math_ops.reduce_logsumexp(
409 math_ops.reduce_logsumexp(
441 return math_ops.reduce_logsumexp(
Dmixture_same_family.py268 return math_ops.reduce_logsumexp(
285 return math_ops.reduce_logsumexp(
Dmoving_stats.py172 x = math_ops.reduce_logsumexp(x, axis=0)
Dmixture.py299 log_sum_exp = math_ops.reduce_logsumexp(concat_log_probs, [0])
312 mixture_log_cdf = math_ops.reduce_logsumexp(concatted_log_cdfs, [0])
Donehot_categorical.py240 math_ops.reduce_logsumexp(x, axis=[-1])),
Drelaxed_onehot_categorical.py304 math_ops.reduce_logsumexp(x, axis=[-1])),
Dconditional_transformed_distribution.py118 return math_ops.reduce_logsumexp(array_ops.stack(lp_on_fibers), axis=0)
Dvector_diffeomixture.py653 return math_ops.reduce_logsumexp(
/external/tensorflow/tensorflow/contrib/labeled_tensor/
D__init__.py126 reduce_logsumexp = _ops.reduce_logsumexp variable
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_ops.py354 self._prior_probs[shard_id] = math_ops.reduce_logsumexp(
450 op.append(math_ops.reduce_logsumexp(prior_probs))
451 self._log_likelihood_op = math_ops.reduce_logsumexp(op)
459 math_ops.reduce_logsumexp(op, axis=2, keepdims=True), axis=0)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dsoftmax_centered.py171 math_ops.reduce_logsumexp(x, axis=-1, keepdims=True))
/external/tensorflow/tensorflow/contrib/nn/python/ops/
Dsampling_ops.py88 return math_ops.reduce_logsumexp(
/external/tensorflow/tensorflow/contrib/crf/python/kernel_tests/
Dcrf_test.py194 brute_force_log_norm = math_ops.reduce_logsumexp(all_sequence_scores)
243 total_log_likelihood = math_ops.reduce_logsumexp(
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
Dmixture_same_family_test.py121 expected_log_cdf = math_ops.reduce_logsumexp(
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops.py969 reduce_logsumexp = define_reduce_op('reduce_logsumexp', variable
970 math_ops.reduce_logsumexp)
/external/tensorflow/tensorflow/python/ops/distributions/
Dtransformed_distribution.py437 return math_ops.reduce_logsumexp(array_ops.stack(lp_on_fibers), axis=0)
Dutil.py1111 lswe = math_ops.reduce_logsumexp(logx, axis=axis, keepdims=keep_dims)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt316 name: "reduce_logsumexp"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt316 name: "reduce_logsumexp"
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dutil_test.py826 expected = math_ops.reduce_logsumexp(logx, axis=-1)

12