Searched defs:_train (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/examples/ |
| D | stateful_example.py | 47 def _train(model, optim, train_steps=1): function
|
| /external/sdv/vsomeip/implementation/endpoints/src/ |
| D | server_endpoint_impl.cpp | 528 const std::shared_ptr<train>& _train, in queue_train() 571 const std::shared_ptr<train>& _train) { in flush() 755 const std::shared_ptr<train>& _train, const boost::system::error_code &_error_code) { in flush_cbk()
|
| /external/pytorch/torch/_export/ |
| D | __init__.py | 202 def _train(self, mode: bool = True): function
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | export_utils.py | 211 def _train(self, mode: bool = True): function
|
| D | utils.py | 622 def _train(self, mode: bool = True): function
|
| /external/pytorch/test/distributed/checkpoint/e2e/ |
| D | test_e2e_save_and_load.py | 122 def _train(model, optim, train_steps=1): function
|
| /external/pytorch/torch/export/ |
| D | exported_program.py | 964 def _train(self, mode: bool = True): function
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 239 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): member in LayoutOptimizerTest
|