Searched refs:sample_inputs_foreach (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 95 sample_inputs_foreach,
9358 … return [sample_inputs_foreach(None, device, dtype, num_tensors, **_foreach_inputs_kwargs)]
9478 …input = sample_inputs_foreach(None, device, dtype, NUM_SIZE0_TENSORS, **zero_size_foreach_inputs_k…
9481 … sample_inputs_foreach(None, device, dtype, NUM_SIZE0_TENSORS, **zero_size_foreach_inputs_kwargs)
9521 input = sample_inputs_foreach(
9526 sample_inputs_foreach(
9574 …input = sample_inputs_foreach(None, device, dtype, NUM_SIZE0_TENSORS, zero_size=True, **_foreach_i…
9592 …input = sample_inputs_foreach(None, device, dtype, num_tensors, zero_size=False, **_foreach_inputs…
9634 …input = sample_inputs_foreach(None, device, dtype, NUM_SIZE0_TENSORS, zero_size=True, **_foreach_i…
9636 …sample_inputs_foreach(None, device, dtype, NUM_SIZE0_TENSORS, zero_size=True, **_foreach_inputs_kw…
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 2721 def sample_inputs_foreach( function
|