Searched refs:_test_sparse_op_input_tensors (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_type_promotion.py | 785 def _test_sparse_op_input_tensors(self, device, dtype, coalesced, zeros=True): member in TestTypePromotion
829 (dense1, sparse1) = self._test_sparse_op_input_tensors(device, dtype1, coalesced)
830 …(dense2, sparse2) = self._test_sparse_op_input_tensors(device, dtype2, coalesced, op_name != 'div')
|