Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py1225 restrict_to_index=OPINFO_SAMPLE_INPUT_INDEX,
1250 restrict_to_index=OPINFO_SAMPLE_INPUT_INDEX,
1270 restrict_to_index=OPINFO_SAMPLE_INPUT_INDEX,
1283 restrict_to_index=OPINFO_SAMPLE_INPUT_INDEX,
1297 restrict_to_index=OPINFO_SAMPLE_INPUT_INDEX,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py363 callback=lambda x: x, set_seed=True, restrict_to_index=None): argument
374 self.restrict_to_index = restrict_to_index
392 if (self.restrict_to_index is None) or (input_idx == self.restrict_to_index):