Searched defs:sample_input (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_save_load_for_op_version.py | 78 def test_versioned_div_tensor(self, sample_input): argument
131 def test_versioned_div_tensor_inplace(self, sample_input): argument
181 def test_versioned_div_tensor_out(self, sample_input): argument
238 def test_versioned_div_scalar(self, sample_input): argument
300 def test_versioned_div_scalar_reciprocal(self, sample_input): argument
371 def test_versioned_div_scalar_inplace(self, sample_input): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_expanded_weights.py | 217 def _compare_ew_and_for_loop_per_sample_grads(self, op, sample_input, reduction): argument
1066 def make_expanded_weight(sample_input, batch_size, loss_reduction="sum"): argument
|
| H A D | test_meta.py | 1326 def sample_input(op, device, dtype, requires_grad, **kwargs): function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 119 def is_valid_inplace_sample_input(sample_input, op, inplace_variant): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 1002 def _serialize_sample(sample_input): argument
|
| H A D | common_utils.py | 3736 def compare_with_reference(self, torch_fn, ref_fn, sample_input, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 331 def __init__(self, sample_input, *, error_type=RuntimeError, error_regex): argument
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 8713 def _test_rnn_impl(self, qconfigs, M, module_type_strs, module_types, sample_input): argument
|