Home
last modified time | relevance | path

Searched refs:foreach_num_tensors (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py9293 foreach_num_tensors = [20, 23] if not TEST_WITH_SLOW else [23, 30, 300] variable
9507 …t_tensors = kwargs.pop("num_input_tensors") if num_input_tensors_specified else foreach_num_tensors
9581 num_input_tensors = kwargs.pop("num_input_tensors", foreach_num_tensors)
9646 …t_tensors = kwargs.pop("num_input_tensors") if num_input_tensors_specified else foreach_num_tensors