def test_sorted_segment_ops(self):
SegmentsTester()._test(
'SortedSegment',
hu.segmented_tensor(dtype=np.float32,
is_sorted=True,
allow_empty=True),
REFERENCES_ALL + REFERENCES_SORTED)(self)
def test_unsorted_segment_ops(self):
SegmentsTester()._test(
'UnsortedSegment',
hu.segmented_tensor(dtype=np.float32,
is_sorted=False,
allow_empty=True),
REFERENCES_ALL,
)(self)
def test_unsorted_segment_ops(self):
SegmentsTester()._test(
'UnsortedSegment',
hu.segmented_tensor(
dtype=np.float32,
is_sorted=False,
allow_empty=True
),
REFERENCES_ALL,
)(self)
def test_sorted_segment_ops(self):
SegmentsTester()._test(
'SortedSegment',
hu.segmented_tensor(
dtype=np.float32,
is_sorted=True,
allow_empty=True
),
REFERENCES_ALL + REFERENCES_SORTED
)(self)
def test_unsorted_segment_ops_gpu(self):
SegmentsTester()._test(
'UnsortedSegment',
hu.segmented_tensor(
dtype=np.float32,
is_sorted=False,
allow_empty=True,
),
REFERENCES_ALL,
gpu=workspace.has_gpu_support,
grad_check=False,
)(self)
def test_unsorted_segment_ops_gpu(self):
SegmentsTester()._test(
'UnsortedSegment',
hu.segmented_tensor(
dtype=np.float32,
is_sorted=False,
allow_empty=True,
),
REFERENCES_ALL,
gpu=workspace.has_gpu_support,
grad_check=False,
)(self)
def test_unsorted_segment_ops(self):
test('UnsortedSegment',
hu.segmented_tensor(dtype=np.float32, is_sorted=False),
REFERENCES_ALL)(self)
def test_sorted_segment_ops(self):
test('SortedSegment',
hu.segmented_tensor(dtype=np.float32, is_sorted=True),
REFERENCES_ALL + REFERENCES_SORTED)(self)
def test_unsorted_segment_ops(self):
test(
'UnsortedSegment',
hu.segmented_tensor(dtype=np.float32, is_sorted=False),
REFERENCES_ALL)(self)
def test_sorted_segment_ops(self):
test(
'SortedSegment',
hu.segmented_tensor(dtype=np.float32, is_sorted=True),
REFERENCES_ALL + REFERENCES_SORTED)(self)
