Searched refs:recompute_ops (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | graph_split.py | 136 def __init__(self, init_op, is_output, unique_id, reach_tab, recompute_ops=None): argument
143 self.recompute_ops = [] if recompute_ops is None else recompute_ops
246 self.recompute_ops.extend(area.recompute_ops)
269 tail_tensor = area.recompute_ops[-1].output
272 if area.recompute_ops[0].inputs:
273 tensor_map[area.recompute_ops[0].inputs[0]] = area.recompute_ops[0].inputs[0]
274 for op in area.recompute_ops:
280 for op in area.recompute_ops:
464 …s.append(Graph('{}_{}'.format(self.graph.name, i), area.ops, area.stitch_info, area.recompute_ops))
520 self.recom_area.recompute_ops.extend(ops)
[all …]
|
| D | model_builder.py | 183 if subgraph.recompute_ops:
184 desc['recompute_ops'] = [op.output.name for op in subgraph.recompute_ops]
228 if op in graph_ops or op in subgraph.recompute_ops:
|
| D | model.py | 388 def __init__(self, name, ops, stitch_info=None, recompute_ops=None): argument
394 self.recompute_ops = recompute_ops
|
| /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
55 return std::set<std::string>(recompute_ops.begin(), recompute_ops.end()); in GetRecomputeOps()
60 bool IsRecomputeOp(const nlohmann::json &op_desc, const std::set<std::string> &recompute_ops) { in IsRecomputeOp() argument
66 if (recompute_ops.count(tensor_name)) { in IsRecomputeOp()
142 auto recompute_ops = GetRecomputeOps(kernel_json); in DecodeSplitNodes() local
161 if (IsRecomputeOp(op_desc, recompute_ops)) { in DecodeSplitNodes()
|