Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py625 label_olabels = math_ops.reduce_logsumexp(label_states + one_hot, axis=2)
626 blank_olabels = math_ops.reduce_logsumexp(blank_states, axis=2, keepdims=True)
670 blank_olabels = math_ops.reduce_logsumexp(blank_states, axis=2, keepdims=True)
1280 return math_ops.reduce_logsumexp(states + one_hot, axis=-1)
1320 state_log_prob = math_ops.reduce_logsumexp(state_log_prob, axis=-1)
1322 log_prob_sum = math_ops.reduce_logsumexp(
1337 state_log_prob = math_ops.reduce_logsumexp(state_log_prob, axis=-1)
1339 log_prob_sum = math_ops.reduce_logsumexp(
1361 fwd_bwd_log_probs_sum = math_ops.reduce_logsumexp(
Dmath_ops_test.py153 y_tf_np = math_ops.reduce_logsumexp(x_np)
161 y_tf = math_ops.reduce_logsumexp(x_np, axis=[0])
171 y_tf = math_ops.reduce_logsumexp(x_np, axis=0)
181 y_tf_np = math_ops.reduce_logsumexp(x_np, keepdims=True)
199 y_tf_np = math_ops.reduce_logsumexp(x_tf)
216 y_tf_np = math_ops.reduce_logsumexp(x_tf)
222 res = math_ops.reduce_logsumexp(-np.inf)
230 y_rt = math_ops.reduce_logsumexp(x_rt)
Dmath_ops.py3386 return reduce_logsumexp(input_tensor, axis, keepdims, name)
3391 def reduce_logsumexp(input_tensor, axis=None, keepdims=False, name=None): function
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py88 self.assertAllClose(tf.reduce_logsumexp(a, reduction_indices=[1]).eval(),
91 tf.reduce_logsumexp(a, reduction_indices=[0, 1]).eval(),
94 tf.reduce_logsumexp(a, [0, 1]).eval(), 6.45619344711)
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dcumulative_logsumexp_test.py95 lambda i: math_ops.reduce_logsumexp(x[:i + 1]),
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py346 return math_ops.exp(inputs - math_ops.reduce_logsumexp(
/external/tensorflow/tensorflow/python/ops/distributions/
Dtransformed_distribution.py433 return math_ops.reduce_logsumexp(array_ops.stack(lp_on_fibers), axis=0)
Dutil.py1114 lswe = math_ops.reduce_logsumexp(logx, axis=axis, keepdims=keep_dims)
/external/tensorflow/tensorflow/python/eager/
Dbenchmarks_test.py916 math_ops.reduce_logsumexp, jit_compile=xla_compile)
919 func = lambda: math_ops.reduce_logsumexp(x)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt356 name: "reduce_logsumexp"
Dtensorflow.pbtxt1912 name: "reduce_logsumexp"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt356 name: "reduce_logsumexp"
Dtensorflow.pbtxt920 name: "reduce_logsumexp"
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dutil_test.py823 expected = math_ops.reduce_logsumexp(logx, axis=-1)
/external/tensorflow/tensorflow/python/keras/
Dbackend.py2628 return math_ops.reduce_logsumexp(x, axis, keepdims)
/external/tensorflow/
DRELEASE.md5294 * Add `reduce_logsumexp` benchmark with experiment compile.
6230 * A few performance optimizations on `tf.reduce_logsumexp`.
9450 `tf.reduce_join`: `reduction_indices` becomes `axis` * `tf.reduce_logsumexp`: