Searched refs:unsorted_segment_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 159 def _ragged_segment_aggregate(unsorted_segment_op, argument
201 return unsorted_segment_op(data, segment_ids, num_segments, separator,
204 return unsorted_segment_op(data, segment_ids, num_segments, name)
224 return _ragged_segment_aggregate(unsorted_segment_op, data.values,
257 output_values = _ragged_segment_aggregate(unsorted_segment_op, data.values,
435 unsorted_segment_op, argument
525 inner_reduced = ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
528 return ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
543 result = _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
553 result = _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
[all …]
|