Home
last modified time | relevance | path

Searched refs:_generate_sample_data (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_maskedtensor.py85 def _generate_sample_data( function
230 for sample in _generate_sample_data(device=device):
243 samples = _generate_sample_data(
246 ) + _generate_sample_data(device=device, layout=torch.sparse_csr)
261 for sample in _generate_sample_data(device=device, layout=torch.strided):
279 for sample in _generate_sample_data(device=device, layout=torch.strided):