Searched refs:predict_map (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_distribute_predict.py | 57 predict_map = model.infer_predict_layout(inputs)
60 return predict_map, output
78 predict_map = None
79 rank_list = _infer_rank_list(train_map, predict_map)["weight"]
87 predict_map = {'weight': [[8], [-1, 0]]}
88 rank_list = _infer_rank_list(train_map, predict_map)
96 predict_map = {'weight': [[4, 8], [-1, 0]]}
97 rank_list = _infer_rank_list(train_map, predict_map)
105 predict_map = {'weight': [[2, 2], [1, 0]]}
106 rank_list = _infer_rank_list(train_map, predict_map)["weight"]
[all …]
|
| D | test_scatter_update.py | 50 predict_map = model.infer_predict_layout(inputs)
53 return predict_map, output
75 predict_map = model.infer_predict_layout(inputs)
78 return predict_map, output
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 313 def _infer_rank_list(train_map, predict_map=None): argument
333 if not predict_map:
336 if param_name not in predict_map:
339 predict_layout = predict_map[param_name]
|