Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py709 sample_inputs_sparse_csc_func: Callable = None variable in OpInfo
1042 self.supports_sparse_csc = self.sample_inputs_sparse_csc_func is not None
1043 if self.sample_inputs_sparse_csc_func is None:
1044 self.sample_inputs_sparse_csc_func = self._sample_inputs_unspecified
1067 self.sample_inputs_sparse_csc_func = torch.no_grad()(
1068 self.sample_inputs_sparse_csc_func
1373 return self.sample_inputs_sparse_csc_func(
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py12010sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_mul, layout=torch.sparse_csc),
18610sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csc),
18658sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csc),
20947sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_reduction_sum, layout=torch.sparse_csc),