Home
last modified time | relevance | path

Searched refs:reduce_prod (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_reduce_op_test.py83 ragged_reduce_op=ragged_math_ops.reduce_prod,
90 ragged_reduce_op=ragged_math_ops.reduce_prod,
185 ragged_reduce_op=ragged_math_ops.reduce_prod,
192 ragged_reduce_op=ragged_math_ops.reduce_prod,
278 ragged_reduce_op=ragged_math_ops.reduce_prod,
309 ragged_reduce_op=ragged_math_ops.reduce_prod,
343 ragged_reduce_op=ragged_math_ops.reduce_prod,
Dragged_math_ops.py579 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): function
582 reduce_op=math_ops.reduce_prod,
642 reduce_prod(_cast(input_tensor, dtypes.int32), axis, keepdims),
664 _set_ragged_reduce_docstring(reduce_prod, 'product', 'multiplied', '1',
Dragged_dispatch.py405 math_ops.reduce_prod,
520 (math_ops.reduce_prod, ragged_math_ops.reduce_prod, ['input_tensor']),
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py64 tf.reduce_prod(
67 tf.reduce_prod(
69 self.assertAllEqual(tf.reduce_prod(a, [0, 1]).eval(), 720.0)
/external/tensorflow/tensorflow/python/ops/
Dbincount_ops.py126 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr)) > 0
163 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr.values)) > 0
165 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr)) > 0
Ddata_flow_grad.py39 math_ops.range(math_ops.reduce_prod(prefix_shape)), prefix_shape)
Dmath_grad_test.py151 outputs = math_ops.reduce_prod(inputs)
162 outputs = math_ops.reduce_prod(inputs, -1)
174 outputs = math_ops.reduce_prod(inputs)
186 outputs = math_ops.reduce_prod(inputs, -1)
Dmath_ops.py2526 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): function
2629 return reduce_prod(input_tensor, axis, keepdims, name)
4712 prod_free_dims = reduce_prod(free_dims)
4713 prod_axes_dims = reduce_prod(axes_dims)
Dmath_grad.py268 math_ops.reduce_prod(input_shape), math_ops.reduce_prod(output_shape))
302 reduced_num = math_ops.reduce_prod(array_ops.gather(input_shape, reduced))
303 other_num = math_ops.reduce_prod(array_ops.gather(input_shape, other))
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dreduction.py37 return math_ops.reduce_prod
Dnormalization.py188 batch_count = math_ops.reduce_prod(batch_reduce_shape)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_circulant.py256 (vec_leading_shape, [math_ops.reduce_prod(vec_block_shape)]), 0)
332 n = math_ops.reduce_prod(trailing_dims)
432 det = math_ops.reduce_prod(self.spectrum, axis=axis)
Dlinear_operator_lower_triangular.py201 return math_ops.reduce_prod(self._get_diag(), axis=[-1])
Dlinear_operator_diag.py235 return math_ops.reduce_prod(self._diag, axis=[-1])
Dlinalg_impl.py993 broadcast_batch_size = math_ops.reduce_prod(broadcast_batch_shape)
1135 batch_size = math_ops.reduce_prod(shape[:-2])
/external/tensorflow/tensorflow/compiler/tests/
Dreduce_ops_test.py110 self._testReduction(math_ops.reduce_prod, np.prod, np.float32,
114 self._testReduction(math_ops.reduce_prod, np.prod, np.complex64,
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py420 return math_ops.reduce_prod(
445 return math_ops.reduce_prod(math_ops.tanh(x)**2)
478 increment = 3. / math_ops.reduce_prod(
628 return math_ops.reduce_prod(math_ops.tanh(x)**2)
656 return math_ops.reduce_prod(math_ops.tanh(x)**2)
666 return math_ops.reduce_prod(math_ops.tanh(x)**2)
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dreduce.py254 return make_reduce_tests(tf.reduce_prod, -2, 2)(options)
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_v2.py161 multiplier = math_ops.cast(math_ops.reduce_prod(axes_vals),
Dlocal.py804 in_size = math_ops.reduce_prod(in_dims)
805 out_size = math_ops.reduce_prod(out_dims)
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduction_ops_test.py107 math_ops.reduce_prod, math_ops.reduce_max,
125 math_ops.reduce_prod, math_ops.reduce_max,
633 return math_ops.reduce_prod(x, reduction_axes, keepdims)
644 v = math_ops.reduce_prod([0, 0], constant_op.constant(0, dtype=dtype))
716 y = math_ops.reduce_prod(x, [1])
726 y = math_ops.reduce_prod(x, [0])
/external/tensorflow/tensorflow/python/ops/distributions/
Dtransformed_distribution.py470 prob = math_ops.reduce_prod(prob, self._reduce_event_indices)
548 entropy *= math_ops.cast(math_ops.reduce_prod(self._override_event_shape),
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py205 return _math_ops.reduce_prod(_array_ops.shape(grad)[-rank:])
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py430 num_elements = [math_ops.reduce_prod(s) for s in shapes]
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Drevive_test.py82 (math_ops.reduce_prod(array_ops.shape(inputs)[:-1]),

123