Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_segment_op_test.py92 (ragged_math_ops.segment_max, max, [0, 0, 1, 1, 2, 2]),
93 (ragged_math_ops.segment_max, max, [0, 0, 0, 1, 1, 1]),
94 (ragged_math_ops.segment_max, max, [5, 4, 3, 2, 1, 0]),
95 (ragged_math_ops.segment_max, max, [0, 0, 0, 10, 10, 10]),
124 (ragged_math_ops.segment_max, max, [0, 0, 1, 1, 2, 2]),
125 (ragged_math_ops.segment_max, max, [0, 0, 0, 1, 1, 1]),
126 (ragged_math_ops.segment_max, max, [5, 4, 3, 2, 1, 0]),
127 (ragged_math_ops.segment_max, max, [0, 0, 0, 10, 10, 10]),
Dragged_math_ops.py309 def segment_max(data: ragged_tensor.RaggedOrDense, function
371 _set_ragged_segment_docstring(segment_max, 'maximum', 'maximized')
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SegmentMax.pbtxt4 name: "math.segment_max"
7 name: "segment_max"
/external/tensorflow/tensorflow/python/ops/
Dmath_grad_test.py343 segment_max = math_ops.segment_max(data, segment_ids)
345 error = gradient_checker.compute_gradient_error(data, [3], segment_max,
365 segment_max = math_ops.segment_max(data, segment_ids)
367 error = gradient_checker.compute_gradient_error(inputs, [1], segment_max,
/external/tensorflow/tensorflow/python/feature_column/
Dutils.py34 seq_length = math_ops.segment_max(column_ids, segment_ids=row_ids)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SegmentMax.pbtxt45 >>> tf.math.segment_max(c, tf.constant([0, 0, 1])).numpy()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-200.md12 tf.math.segment_max(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_deterministic_test.py56 math_ops.segment_max: False,
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dsegment_reduction_ops_d9m_test.py64 math_ops.segment_max: False,
Dsegment_reduction_ops_test.py98 (np.maximum, None, math_ops.segment_max)]
245 math_ops.segment_max
272 math_ops.segment_max,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt404 name: "segment_max"
Dtensorflow.pbtxt2056 name: "segment_max"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt404 name: "segment_max"
/external/tensorflow/
DRELEASE.md8015 `tf.math.rsqrt`, `tf.math.segment_max`, `tf.math.segment_mean`,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td14530 >>> tf.math.segment_max(c, tf.constant([0, 0, 1])).numpy()