Searched refs:_segmentReduction (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | segment_reduction_ops_test.py | 35 def _segmentReduction(self, op, data, indices, num_segments): member in SegmentReductionOpsTest
45 return self._segmentReduction(math_ops.unsorted_segment_sum, data, indices,
49 return self._segmentReduction(math_ops.unsorted_segment_prod, data, indices,
53 return self._segmentReduction(math_ops.unsorted_segment_min, data, indices,
57 return self._segmentReduction(math_ops.unsorted_segment_max, data, indices,
158 functools.partial(self._segmentReduction,
|