Searched defs:TrainingMode (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_symbolic_helper.py | 35 self, op_train_mode: int, export_mode: torch.onnx.TrainingMode
58 export_mode: torch.onnx.TrainingMode,
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | utils.py | 62 def select_model_mode_for_export(model, mode: _C_onnx.TrainingMode):
153 def exporter_context(model, mode: _C_onnx.TrainingMode, verbose: bool):
183 training: _C_onnx.TrainingMode = _C_onnx.TrainingMode.EVAL,
1289 training: _C_onnx.TrainingMode = _C_onnx.TrainingMode.EVAL,
|
| H A D | _globals.py | 42 def training_mode(self, training_mode: _C_onnx.TrainingMode): member in _InternalGlobals
|
| H A D | __init__.py | 137 def export(
|
| H A D | verification.py | 781 training: _C_onnx.TrainingMode = _C_onnx.TrainingMode.EVAL,
1683 training: _C_onnx.TrainingMode = _C_onnx.TrainingMode.EVAL,
|
| /aosp_15_r20/external/pytorch/torch/csrc/onnx/ |
| H A D | onnx.h | 12 enum class TrainingMode { enum
|
| /aosp_15_r20/external/aws-sdk-java-v2/services/personalize/src/main/resources/codegen-resources/ |
| H A D | service-2.json | 4927 "TrainingMode":{ object
|