コード例 #1
0
 def test_sparse_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True
         ),
         REFERENCES_ALL
     )(self)
コード例 #2
0
ファイル: segment_ops_test.py プロジェクト: gtgalone/pytorch
 def test_sparse_unsorted_segment_ops(self):
     SegmentsTester()._test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(
             dtype=np.float32,
             is_sorted=False,
             allow_empty=True
         ),
         REFERENCES_ALL
     )(self)
コード例 #3
0
ファイル: segment_ops_test.py プロジェクト: zxsted/caffe2
 def test_sparse_unsorted_segment_ops(self):
     test('SparseUnsortedSegment',
          hu.sparse_segmented_tensor(dtype=np.float32, is_sorted=False),
          REFERENCES_ALL)(self)
コード例 #4
0
 def test_sparse_unsorted_segment_ops(self):
     test(
         'SparseUnsortedSegment',
         hu.sparse_segmented_tensor(dtype=np.float32, is_sorted=False),
         REFERENCES_ALL)(self)