Searched defs:recompute_ops (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_splitter.cc | 54 std::vector<std::string> recompute_ops = kernel_json[kJsonKeyRecomputeOps]; in GetRecomputeOps() local
60 bool IsRecomputeOp(const nlohmann::json &op_desc, const std::set<std::string> &recompute_ops) { in IsRecomputeOp()
142 auto recompute_ops = GetRecomputeOps(kernel_json); in DecodeSplitNodes() local
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | model.py | 388 def __init__(self, name, ops, stitch_info=None, recompute_ops=None): argument
|
| D | graph_split.py | 136 def __init__(self, init_op, is_output, unique_id, reach_tab, recompute_ops=None): argument
|