Searched defs:device_capa (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torchgen/_autoheuristic/ |
| D | train.py | 142 def gen_precondition(self, opt_name, shared_memory, device_capa): argument
151 self, heuristic_name, opt_name, threshold, shared_memory, device_capa, dt argument
|
| D | train_regression.py | 438 self, heuristic_name, opt_name, threshold, shared_memory, device_capa, classes argument
|
| D | train_decision.py | 537 self, heuristic_name, opt_name, threshold, shared_memory, device_capa, classes argument
|