| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | fully_sharded_data_parallel.py | 1017 def _assert_state(self, state: Union[TrainingState, List[TrainingState]]) -> None:
2096 self, training_state: TrainingState, handle_training_state: HandleTrainingState
|
| H A D | _common_utils.py | 178 class TrainingState(Enum): class
447 training_states: List[TrainingState],
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
| H A D | _fsdp_param_group.py | 76 self, training_state: TrainingState
483 def use_training_state(self, training_state: TrainingState):
|
| H A D | _fsdp_common.py | 59 class TrainingState(Enum): class
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2/src/main/proto/google/cloud/retail/v2/ |
| H A D | model.proto | 70 enum TrainingState { enum
|
| /aosp_15_r20/external/googleapis/google/cloud/retail/v2/ |
| H A D | model.proto | 70 enum TrainingState { enum
|
| /aosp_15_r20/external/googleapis/google/cloud/retail/v2beta/ |
| H A D | model.proto | 70 enum TrainingState { enum
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2beta/src/main/proto/google/cloud/retail/v2beta/ |
| H A D | model.proto | 70 enum TrainingState { enum
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2alpha/src/main/proto/google/cloud/retail/v2alpha/ |
| H A D | model.proto | 273 enum TrainingState { enum
|
| /aosp_15_r20/external/googleapis/google/cloud/retail/v2alpha/ |
| H A D | model.proto | 273 enum TrainingState { enum
|
| /aosp_15_r20/external/googleapis/google/cloud/discoveryengine/v1alpha/ |
| H A D | engine.proto | 74 enum TrainingState { enum
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2/src/main/java/com/google/cloud/retail/v2/ |
| H A D | Model.java | 273 public enum TrainingState implements com.google.protobuf.ProtocolMessageEnum { enum in Model
414 private TrainingState(int value) { in TrainingState() method in Model.TrainingState
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2beta/src/main/java/com/google/cloud/retail/v2beta/ |
| H A D | Model.java | 274 public enum TrainingState implements com.google.protobuf.ProtocolMessageEnum { enum in Model
415 private TrainingState(int value) { in TrainingState() method in Model.TrainingState
|
| /aosp_15_r20/external/google-cloud-java/java-retail/proto-google-cloud-retail-v2alpha/src/main/java/com/google/cloud/retail/v2alpha/ |
| H A D | Model.java | 274 public enum TrainingState implements com.google.protobuf.ProtocolMessageEnum { enum in Model
415 private TrainingState(int value) { in TrainingState() method in Model.TrainingState
|