Home
last modified time | relevance | path

Searched defs:_train (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/distributed/checkpoint/examples/
Dstateful_example.py47 def _train(model, optim, train_steps=1): function
/external/sdv/vsomeip/implementation/endpoints/src/
Dserver_endpoint_impl.cpp528 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__.py202 def _train(self, mode: bool = True): function
/external/pytorch/torch/ao/quantization/pt2e/
Dexport_utils.py211 def _train(self, mode: bool = True): function
Dutils.py622 def _train(self, mode: bool = True): function
/external/pytorch/test/distributed/checkpoint/e2e/
Dtest_e2e_save_and_load.py122 def _train(model, optim, train_steps=1): function
/external/pytorch/torch/export/
Dexported_program.py964 def _train(self, mode: bool = True): function
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py239 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): member in LayoutOptimizerTest