Searched refs:_ragged_segment_aggregate (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, function
203 return _ragged_segment_aggregate(unsorted_segment_op, data.values,
238 output_values = _ragged_segment_aggregate(unsorted_segment_op, data.values,
247 return _ragged_segment_aggregate(math_ops.unsorted_segment_sum, data,
254 return _ragged_segment_aggregate(math_ops.unsorted_segment_prod, data,
261 return _ragged_segment_aggregate(math_ops.unsorted_segment_min, data,
268 return _ragged_segment_aggregate(math_ops.unsorted_segment_max, data,
484 return _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
491 return _ragged_segment_aggregate(unsorted_segment_op, rt_input.values,
|