Searched refs:reference_reduction_numpy (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | _masked.py | 32 from torch.testing._internal.opinfo.utils import prod_numpy, reference_reduction_numpy
240 ref = reference_reduction_numpy(numpy_fn)
431 ref=reference_reduction_numpy(np.sum),
642 ref=reference_reduction_numpy(np.amax),
681 ref=reference_reduction_numpy(np.amin),
717 ref=reference_reduction_numpy(np.argmax, supports_keepdims=False),
742 ref=reference_reduction_numpy(np.argmin, supports_keepdims=False),
763 ref=reference_reduction_numpy(np.mean)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | utils.py | 190 def reference_reduction_numpy(f, supports_keepdims=True): function
272 fn = reference_reduction_numpy(np.prod)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 114 reference_reduction_numpy,
11488 g = reference_reduction_numpy(f)
20657 ref=reference_reduction_numpy(np.all),
20669 ref=reference_reduction_numpy(np.any),
20682 ref=reference_reduction_numpy(np.amax),
20697 ref=reference_reduction_numpy(np.amin),
20712 ref=reference_reduction_numpy(np.argmax, supports_keepdims=False),
20720 ref=reference_reduction_numpy(np.argmin, supports_keepdims=False),
20730 ref=reference_reduction_numpy(np.count_nonzero),
20757 ref=reference_reduction_numpy(np.mean),
[all …]
|