Searched refs:gen_predict_fn_def (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/_autoheuristic/ | ||
| H A D | train.py | 155 def gen_predict_fn_def(self): member in AHTrain |
| H A D | train_regression.py | 434 def gen_predict_fn_def(self): member in AHTrainRegressionTree |
| H A D | train_decision.py | 530 def gen_predict_fn_def(self): member in AHTrainDecisionTree |