Searched defs:model_tp (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/ |
| H A D | test_tp_examples.py | 216 def _setup_optimizer(self, model, model_tp): argument
224 self, model, model_tp, inp, expected_comms_dict=None, check_comms=True argument
238 model_tp, argument
255 model_tp, argument
272 def _thaw_params(thaw_params, model, model_tp): argument
|