Lines Matching refs:sample_input

233                 for sample_input in sample_inputs_func(device, dtype):
234                     mask = sample_input.kwargs.get('mask')
236                         yield sample_input
238                         if layout == sample_input.input.layout:
239                             yield sample_input
241                             sample_input_kwargs = sample_input.kwargs.copy()
243                             yield SampleInput(sample_input.input.clone(),
244                                               args=sample_input.args,
247                             sample_input_kwargs = sample_input.kwargs.copy()
249                             yield SampleInput(sample_input.input.clone(),
250                                               args=sample_input.args,
252 …         if layout != torch.sparse_csr and op.supports_sparse_csr and sample_input.input.ndim == 2:
253                             sample_input_kwargs = sample_input.kwargs.copy()
255                             yield SampleInput(sample_input.input.clone(),
256                                               args=sample_input.args,
282         for sample_input in sample_inputs:
283             t_inp, t_args, t_kwargs = sample_input.input, sample_input.args, sample_input.kwargs