Home
last modified time | relevance | path

Searched refs:segment_prod (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_segment_op_test.py88 (ragged_math_ops.segment_prod, prod, [0, 0, 1, 1, 2, 2]),
89 (ragged_math_ops.segment_prod, prod, [0, 0, 0, 1, 1, 1]),
90 (ragged_math_ops.segment_prod, prod, [5, 4, 3, 2, 1, 0]),
91 (ragged_math_ops.segment_prod, prod, [0, 0, 0, 10, 10, 10]),
120 (ragged_math_ops.segment_prod, prod, [0, 0, 1, 1, 2, 2]),
121 (ragged_math_ops.segment_prod, prod, [0, 0, 0, 1, 1, 1]),
122 (ragged_math_ops.segment_prod, prod, [5, 4, 3, 2, 1, 0]),
123 (ragged_math_ops.segment_prod, prod, [0, 0, 0, 10, 10, 10]),
Dragged_math_ops.py274 def segment_prod(data, segment_ids, num_segments, name=None): function
343 _set_ragged_segment_docstring(segment_prod, 'product', 'multiplied')
Dragged_dispatch.py507 (math_ops.unsorted_segment_prod, ragged_math_ops.segment_prod,
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SegmentProd.pbtxt4 name: "math.segment_prod"
7 name: "segment_prod"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SegmentProd.pbtxt37 tf.segment_prod(c, tf.constant([0, 0, 1]))
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py100 (np.ndarray.__mul__, None, math_ops.segment_prod),
106 (np.ndarray.__mul__, None, math_ops.segment_prod),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt408 name: "segment_prod"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt408 name: "segment_prod"
Dtensorflow.pbtxt2060 name: "segment_prod"
/external/tensorflow/tensorflow/python/ops/
Dmath_grad_test.py383 return math_ops.segment_prod(x, segment_ids)
Dmath_grad.py445 non_zero_prod = gen_math_ops.segment_prod(non_zero_data, segment_ids)
/external/tensorflow/
DRELEASE.md3032 …segment_max`, `tf.math.segment_mean`, `tf.math.segment_min`, `tf.math.segment_prod`, `tf.math.segm…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td13117 tf.segment_prod(c, tf.constant([0, 0, 1]))