HomeSort by relevance Sort by last modified time
    Searched refs:gen_math_ops (Results 1 - 19 of 19) sorted by null

  /external/tensorflow/tensorflow/contrib/quantization/python/
math_ops.py 24 from tensorflow.python.ops import gen_math_ops
25 from tensorflow.python.ops.gen_math_ops import *
  /external/tensorflow/tensorflow/python/ops/
math_ops.py 166 from tensorflow.python.ops import gen_math_ops
174 from tensorflow.python.ops.gen_math_ops import *
181 linspace = gen_math_ops.lin_space
201 gen_math_ops.arg_max.__doc__.replace("dimensions", "axes").replace(
214 return gen_math_ops.arg_max(input, axis, name=name, output_type=output_type)
221 gen_math_ops.arg_min.__doc__.replace("dimensions", "axes").replace(
234 return gen_math_ops.arg_min(input, axis, name=name, output_type=output_type)
269 x_abs = gen_math_ops._complex_abs(
273 x_abs = gen_math_ops._abs(x.values, name=name)
279 return gen_math_ops._complex_abs(x, Tout=x.dtype.real_dtype, name=name
    [all...]
histogram_ops.py 32 from tensorflow.python.ops import gen_math_ops
144 return gen_math_ops._histogram_fixed_width( # pylint: disable=protected-access
math_grad.py 29 from tensorflow.python.ops import gen_math_ops
340 num_zeros = gen_math_ops.unsorted_segment_sum(
349 non_zero_prod = gen_math_ops.unsorted_segment_prod(
386 return gen_math_ops._reciprocal_grad(y, grad)
394 return gen_math_ops._reciprocal_grad(y, grad)
405 return cg * -2.0 * b * ca, gen_math_ops._reciprocal_grad(ca, grad)
416 return cg * -2.0 * b * ca, gen_math_ops._reciprocal_grad(ca, grad)
432 return gen_math_ops._sqrt_grad(y, grad)
450 return gen_math_ops._rsqrt_grad(y, grad)
464 grad_b = gen_math_ops._rsqrt_grad(ca, grad
    [all...]
array_ops.py 103 from tensorflow.python.ops import gen_math_ops
307 return gen_math_ops.cast(input.dense_shape, out_type)
392 return gen_math_ops._prod(
393 gen_math_ops.cast(input.dense_shape, out_type), 0, name=name)
570 packed_begin = gen_math_ops.cast(packed_begin, dtypes.int64)
572 packed_end = gen_math_ops.cast(packed_end, dtypes.int64)
574 packed_strides = gen_math_ops.cast(packed_strides, dtypes.int64)
    [all...]
  /external/tensorflow/tensorflow/contrib/nn/python/ops/
alpha_dropout.py 26 from tensorflow.python.ops import gen_math_ops
76 kept_idx = gen_math_ops.greater_equal(random_tensor, 1 - keep_prob)
  /external/tensorflow/tensorflow/contrib/framework/python/ops/
accumulate_n_v2.py 26 from tensorflow.python.ops import gen_math_ops
103 return gen_math_ops._accumulate_nv2(inputs, name=name, shape=shape)
  /external/tensorflow/tensorflow/contrib/opt/python/training/
drop_stale_gradient_optimizer.py 26 from tensorflow.python.ops import gen_math_ops
108 gen_math_ops.less_equal(staleness, self._staleness),
  /external/tensorflow/tensorflow/contrib/receptive_field/python/util/
graph_compute_order_test.py 26 from tensorflow.python.ops import gen_math_ops
55 gen_math_ops.add(l5, l6, name='L7_add')
parse_layer_parameters_test.py 27 from tensorflow.python.ops import gen_math_ops
57 gen_math_ops.add(l5, l6, name='L7_add')
receptive_field_test.py 29 from tensorflow.python.ops import gen_math_ops
207 gen_math_ops.equal(l1_shape[1], 2), [l1_shape], summarize=4)
  /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
ar_model.py 35 from tensorflow.python.ops import gen_math_ops
132 predicted_covariance = gen_math_ops.exp(log_sigma_square)
221 sigma = math_ops.sqrt(gen_math_ops.maximum(covariance, 1e-5))
296 offset = self.output_window_size * gen_math_ops.minimum(
632 prediction_ops_dict["anomaly_params"] = gen_math_ops.exp(anomaly_log_param)
645 gen_math_ops.maximum(anomaly_variance, 1e-5))
663 sigma = math_ops.sqrt(gen_math_ops.maximum(covariance, 1e-5))
672 p1 = gen_math_ops.minimum(log_prob1, log_prob2)
673 p2 = gen_math_ops.maximum(log_prob1, log_prob2)
674 mixed_log_prob = p2 + math_ops.log(1 + gen_math_ops.exp(p1 - p2)
    [all...]
math_utils.py 36 from tensorflow.python.ops import gen_math_ops
64 new_diagonal = gen_math_ops.maximum(
246 tf.matmul(batch, array_ops.tile(gen_math_ops.expand_dims(matrix, 0),
365 sign_magnitude = (gen_math_ops.exp(
372 diagonal=gen_math_ops.exp((diagonal + _right_pad_with_ones(
723 return stat_variables._replace(start_time=gen_math_ops.minimum(
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
binary_ops_test.py 28 from tensorflow.python.ops import gen_math_ops
74 gen_math_ops._real_div,
111 gen_math_ops._reciprocal_grad,
117 gen_math_ops._sigmoid_grad,
123 gen_math_ops._rsqrt_grad,
129 gen_math_ops._sqrt_grad,
149 gen_math_ops._tanh_grad,
210 gen_math_ops._truncate_div,
372 gen_math_ops._real_div,
381 gen_math_ops._real_div
    [all...]
  /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
periodic.py 29 from tensorflow.python.ops import gen_math_ops
267 math_ops.abs(value - gen_math_ops.round(value)),
331 gen_math_ops.round(original_matrix_powers), dtypes.int64)),
state_space_model.py 40 from tensorflow.python.ops import gen_math_ops
    [all...]
  /external/tensorflow/tensorflow/python/eager/
benchmarks_test.py 44 from tensorflow.python.ops import gen_math_ops
246 gen_math_ops._mat_mul(m, m, transpose_b=transpose_b)
  /external/tensorflow/tensorflow/python/grappler/
layout_optimizer_test.py 38 from tensorflow.python.ops import gen_math_ops
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
cwise_ops_test.py 31 from tensorflow.python.ops import gen_math_ops
498 (gen_math_ops._reciprocal_grad, [-2, 2]),
499 (gen_math_ops._rsqrt_grad, [0.1, 3]),
500 (gen_math_ops._sigmoid_grad, [-2, 2]),
501 (gen_math_ops._sqrt_grad, [0.1, 3]),
502 (gen_math_ops._tanh_grad, [-2, 2]),
    [all...]

Completed in 206 milliseconds