Home
last modified time | relevance | path

Searched defs:with_keepdim (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_reductions.py1698 exact_dtype=True, with_keepdim=False): argument
1742 …def _test_sum_reduction_vs_numpy(self, torch_fn, np_fn, device, dtype, with_keepdim=False, with_ex… argument