Lines Matching refs:math_ops
24 from tensorflow.python.ops import math_ops
52 z = logits - math_ops.reduce_mean(logits, axis=1)[:, array_ops.newaxis]
58 z_cumsum = math_ops.cumsum(z_sorted, axis=1)
59 k = math_ops.range(
60 1, math_ops.cast(dims, logits.dtype) + 1, dtype=logits.dtype)
64 k_z = math_ops.reduce_sum(math_ops.cast(z_check, dtypes.int32), axis=1)
67 indices = array_ops.stack([math_ops.range(0, obs), k_z - 1], axis=1)
69 tau_z = (tau_sum - 1) / math_ops.cast(k_z, logits.dtype)
72 return math_ops.maximum(
73 math_ops.cast(0, logits.dtype), z - tau_z[:, array_ops.newaxis])
