Searched refs:sample_inputs_softmax_variant (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | _masked.py | 36 def sample_inputs_softmax_variant( function
344 for sample_input in sample_inputs_softmax_variant(
361 for sample_input in sample_inputs_softmax_variant(
417 for sample_input in sample_inputs_softmax_variant(
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 138 sample_inputs_softmax_variant,
14364 sample_inputs_func=sample_inputs_softmax_variant,
14375 sample_inputs_func=partial(sample_inputs_softmax_variant, with_dtype=True),
14401 sample_inputs_func=sample_inputs_softmax_variant,
14411 sample_inputs_func=partial(sample_inputs_softmax_variant, with_dtype=True),
19978 sample_inputs_func=sample_inputs_softmax_variant,
19988 sample_inputs_func=partial(sample_inputs_softmax_variant, with_dtype=True),
|