Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_segment_op_test.py96 (ragged_math_ops.segment_max, max, [0, 0, 1, 1, 2, 2]),
97 (ragged_math_ops.segment_max, max, [0, 0, 0, 1, 1, 1]),
98 (ragged_math_ops.segment_max, max, [5, 4, 3, 2, 1, 0]),
99 (ragged_math_ops.segment_max, max, [0, 0, 0, 10, 10, 10]),
128 (ragged_math_ops.segment_max, max, [0, 0, 1, 1, 2, 2]),
129 (ragged_math_ops.segment_max, max, [0, 0, 0, 1, 1, 1]),
130 (ragged_math_ops.segment_max, max, [5, 4, 3, 2, 1, 0]),
131 (ragged_math_ops.segment_max, max, [0, 0, 0, 10, 10, 10]),
Dragged_math_ops.py294 def segment_max(data, segment_ids, num_segments, name=None): function
345 _set_ragged_segment_docstring(segment_max, 'maximum', 'maximized')
Dragged_dispatch.py511 (math_ops.unsorted_segment_max, ragged_math_ops.segment_max,
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SegmentMax.pbtxt4 name: "math.segment_max"
7 name: "segment_max"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SegmentMax.pbtxt37 tf.segment_max(c, tf.constant([0, 0, 1]))
/external/tensorflow/tensorflow/python/ops/
Dmath_grad_test.py348 segment_max = math_ops.segment_max(data, segment_ids)
350 error = gradient_checker.compute_gradient_error(data, [3], segment_max,
370 segment_max = math_ops.segment_max(data, segment_ids)
372 error = gradient_checker.compute_gradient_error(inputs, [1], segment_max,
/external/tensorflow/tensorflow/python/feature_column/
Dutils.py38 seq_length = math_ops.segment_max(column_ids, segment_ids=row_ids)
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py102 (np.maximum, None, math_ops.segment_max)]
246 math_ops.segment_max
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt396 name: "segment_max"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt396 name: "segment_max"
Dtensorflow.pbtxt2048 name: "segment_max"
/external/tensorflow/
DRELEASE.md3032 …amma`, `tf.math.reciprocal`, `tf.math.rint`, `tf.math.rsqrt`, `tf.math.segment_max`, `tf.math.segm…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td12987 tf.segment_max(c, tf.constant([0, 0, 1]))