Searched refs:segment_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_segment_op_test.py | 105 def testRaggedSegment_Int(self, segment_op, combiner, segment_ids): argument
112 segmented = segment_op(rt, segment_ids, num_segments)
141 def testRaggedSegment_Float(self, segment_op, combiner, segment_ids): argument
148 segmented = segment_op(rt, segment_ids, num_segments)
|