Searched refs:reduce_prod (Results 1 – 25 of 56) sorted by relevance
123
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_reduce_op_test.py | 87 ragged_reduce_op=ragged_math_ops.reduce_prod,
94 ragged_reduce_op=ragged_math_ops.reduce_prod,
201 ragged_reduce_op=ragged_math_ops.reduce_prod,
208 ragged_reduce_op=ragged_math_ops.reduce_prod,
306 ragged_reduce_op=ragged_math_ops.reduce_prod,
349 ragged_reduce_op=ragged_math_ops.reduce_prod,
396 ragged_reduce_op=ragged_math_ops.reduce_prod,
|
| D | ragged_math_ops.py | 625 @dispatch.dispatch_for_api(math_ops.reduce_prod)
626 def reduce_prod(input_tensor: ragged_tensor.Ragged, function
632 reduce_op=math_ops.reduce_prod,
741 reduce_prod(_cast(input_tensor, dtypes.int32), axis, keepdims),
767 _set_ragged_reduce_docstring(reduce_prod, 'product', 'multiplied', '1',
|
| D | ragged_dispatch.py | 91 math_ops.reduce_prod,
|
| D | ragged_matmul_op_test.py | 38 values = math_ops.range(math_ops.reduce_prod(shape))
|
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| D | test_file_v0_11.py | 61 tf.reduce_prod(
64 tf.reduce_prod(
66 self.assertAllEqual(tf.reduce_prod(a, [0, 1]).eval(), 720.0)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | bincount_ops.py | 122 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr)) > 0
161 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr.values)) > 0
163 array_is_nonempty = math_ops.reduce_prod(array_ops.shape(arr)) > 0
|
| D | data_flow_grad.py | 35 math_ops.range(math_ops.reduce_prod(prefix_shape)), prefix_shape)
|
| D | math_grad_test.py | 146 outputs = math_ops.reduce_prod(inputs)
157 outputs = math_ops.reduce_prod(inputs, -1)
169 outputs = math_ops.reduce_prod(inputs)
181 outputs = math_ops.reduce_prod(inputs, -1)
|
| D | math_grad.py | 261 math_ops.reduce_prod(input_shape), math_ops.reduce_prod(output_shape))
295 reduced_num = math_ops.reduce_prod(array_ops.gather(input_shape, reduced))
296 other_num = math_ops.reduce_prod(array_ops.gather(input_shape, other))
|
| D | stateful_random_ops.py | 632 delta = math_ops.reduce_prod(shape)
|
| D | embedding_ops.py | 898 math_ops.reduce_prod(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 438 return math_ops.reduce_prod(
461 return math_ops.reduce_prod(
489 return math_ops.reduce_prod(math_ops.tanh(x)**2)
522 increment = 3. / math_ops.reduce_prod(
672 return math_ops.reduce_prod(math_ops.tanh(x)**2)
700 return math_ops.reduce_prod(math_ops.tanh(x)**2)
710 return math_ops.reduce_prod(math_ops.tanh(x)**2)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v1.py | 393 output_batch_size = math_ops.reduce_prod(feature.output_shape).numpy()
430 output_batch_size = math_ops.reduce_prod(feature.output_shape).numpy()
|
| D | tpu_embedding_for_serving.py | 389 output_batch_size = math_ops.reduce_prod(feature.output_shape)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_ops_test.py | 106 self._testReduction(math_ops.reduce_prod, np.prod, np.float32,
110 self._testReduction(math_ops.reduce_prod, np.prod, np.complex64,
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 364 (x_leading_shape, [math_ops.reduce_prod(x_block_shape)]), 0)
455 n = math_ops.reduce_prod(trailing_dims)
570 det = math_ops.reduce_prod(self.spectrum, axis=axis)
|
| D | linear_operator_lower_triangular.py | 202 return math_ops.reduce_prod(self._get_diag(), axis=[-1])
|
| D | linear_operator_diag.py | 230 return math_ops.reduce_prod(self._diag, axis=[-1])
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | reduce.py | 250 return make_reduce_tests(tf.reduce_prod, -2, 2)(options)
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | reduction_ops_test.py | 103 math_ops.reduce_prod, math_ops.reduce_max,
121 math_ops.reduce_prod, math_ops.reduce_max,
629 return math_ops.reduce_prod(x, reduction_axes, keepdims)
640 v = math_ops.reduce_prod([0, 0], constant_op.constant(0, dtype=dtype))
719 y = math_ops.reduce_prod(x, [1])
729 y = math_ops.reduce_prod(x, [0])
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | transformed_distribution.py | 466 prob = math_ops.reduce_prod(prob, self._reduce_event_indices)
544 entropy *= math_ops.cast(math_ops.reduce_prod(self._override_event_shape),
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | build_def.bzl | 125 "reduce_prod",
252 "reduce_prod",
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 205 return _math_ops.reduce_prod(_array_ops.shape(grad)[-rank:])
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 148 [batch_size, math_ops.reduce_prod(x_shape[1:])], axis=0))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 427 num_elements = [math_ops.reduce_prod(s) for s in shapes]
|
123