Searched defs:train_steps (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/examples/ |
| H A D | stateful_example.py | 47 def _train(model, optim, train_steps=1): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/profiler/ |
| H A D | model_analyzer_test.py | 578 def _trainLoop(self, train_op, train_steps, time_dir, time_step, memory_dir, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy_test.py | 307 def train_steps(iterator, steps): function
1117 def train_steps(v, iterator, steps): function
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/e2e/ |
| H A D | test_e2e_save_and_load.py | 122 def _train(model, optim, train_steps=1): argument
|
| /aosp_15_r20/external/googleapis/google/cloud/documentai/v1/ |
| H A D | document_processor_service.proto | 940 int32 train_steps = 2 [(google.api.field_behavior) = OPTIONAL]; field
|
| /aosp_15_r20/external/googleapis/google/cloud/documentai/v1beta3/ |
| H A D | document_processor_service.proto | 1082 int32 train_steps = 2 [(google.api.field_behavior) = OPTIONAL]; field
|