Searched refs:unsorted_segment_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 149 def _ragged_segment_aggregate(unsorted_segment_op, argument
186 return unsorted_segment_op(data, segment_ids, num_segments, name)
203 return _ragged_segment_aggregate(unsorted_segment_op, data.values,
238 output_values = _ragged_segment_aggregate(unsorted_segment_op, data.values,
400 unsorted_segment_op, argument
469 inner_reduced = _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
471 return _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
484 return _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
491 return _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
497 _ragged_reduce_aggregate(reduce_op, unsorted_segment_op,
|