Searched refs:ModuleErrorEnum (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 183 class ModuleErrorEnum(Enum): class
199 error_on=ModuleErrorEnum.CONSTRUCTION_ERROR,
3189 error_on=ModuleErrorEnum.FORWARD_ERROR,
3198 error_on=ModuleErrorEnum.FORWARD_ERROR,
3207 error_on=ModuleErrorEnum.FORWARD_ERROR,
3216 error_on=ModuleErrorEnum.FORWARD_ERROR,
3225 error_on=ModuleErrorEnum.FORWARD_ERROR,
3234 error_on=ModuleErrorEnum.FORWARD_ERROR,
3249 error_on=ModuleErrorEnum.FORWARD_ERROR,
3258 error_on=ModuleErrorEnum.FORWARD_ERROR,
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_modules.py | 15 from torch.testing._internal.common_modules import module_db, modules, ModuleErrorEnum, TrainEvalMo…
854 if error_input.error_on == ModuleErrorEnum.CONSTRUCTION_ERROR:
857 elif error_input.error_on == ModuleErrorEnum.FORWARD_ERROR:
|