| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 813 bool TopCellInfo::IsSubCell(const std::string &cell_id) const { in IsSubCell()
818 if (sub_cell_list_.find(cell_id) != sub_cell_list_.end()) { in IsSubCell()
1728 MS_LOG(DEBUG) << "Top cell " << top_cell_->cell_id() << " run firstly"; in UpdateForwardTensorInfoInBpropGraph()
1730 …S_LOG(EXCEPTION) << "The cell stack is empty when running a new top cell " << top_cell_->cell_id(); in UpdateForwardTensorInfoInBpropGraph()
1739 << top_cell_->cell_id(); in UpdateForwardTensorInfoInBpropGraph()
1784 << top_cell()->cell_id(); in SaveForwardTensorInfoInBpropGraph()
2019 void GradExecutor::PushCellStack(const std::string &cell_id) { cell_stack_.push(cell_id); } in PushCellStack() argument
2047 auto cell_id = GetId(cell); in GetCellId() local
2058 cell_id += "_" + shape->ToString(); in GetCellId()
2059 cell_id += type->ToString(); in GetCellId()
[all …]
|
| D | pynative_execute.h | 51 std::string cell_id; member
56 explicit GraphInfo(std::string id) : cell_id(std::move((id))) {} in GraphInfo()
92 const std::string &cell_id() const { return cell_id_; } in cell_id() function
100 bool IsSubCell(const std::string &cell_id) const;
219 void ClearCellRes(const std::string &cell_id = "");
230 void PushCellStack(const std::string &cell_id);
234 void InitResourceAndDfBuilder(const std::string &cell_id, const py::args &args);
235 void MakeNewTopGraph(const string &cell_id, const py::args &args, bool is_topest);
238 bool IsBpropGraph(const std::string &cell_id);
244 std::string GetAlreadyRunCellId(const std::string &cell_id);
[all …]
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | debug_ops.py | 356 def __init__(self, hook_fn, cell_id=""): argument
359 self.add_prim_attr("cell_id", cell_id)
360 self.init_attrs["cell_id"] = cell_id
365 self.cell_id = cell_id
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | primitive_py.cc | 219 auto cell_id = GetValue<std::string>(this->GetAttr(kCellIDAttrName)); in RunCellHookFunction() local
220 auto iter = hook_grad_.find(cell_id); in RunCellHookFunction()
234 hook_args[0] = cell_id; in RunCellHookFunction()
242 (void)hook_grad_.erase(cell_id); in RunCellHookFunction()
244 hook_grad_[cell_id] = py_args[grad_output_index]; in RunCellHookFunction()
|
| /third_party/mindspore/tests/st/hook/ |
| D | test_hook_function.py | 93 def cell_hook_function(cell_id, grad_input, grad_output): argument
94 print("cell id:", cell_id)
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_hook.py | 54 def cell_hook_function(cell_id, grad_input, grad_output): argument
55 print(cell_id)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kprim.cc | 671 auto cell_id = GetValue<std::string>(prim->GetAttr("cell_id")); in BpropCut() local
672 if (cell_id != "") { in BpropCut()
674 (void)bprop_cut->AddAttr("cell_id", MakeValue(cell_id)); in BpropCut()
|
| D | kpynative.cc | 905 auto cell_id = GetValue<std::string>(prim->GetAttr("cell_id")); in BuildBPropCutFuncGraph() local
906 if (cell_id != "") { in BuildBPropCutFuncGraph()
908 (void)bprop_cut->AddAttr("cell_id", MakeValue(cell_id)); in BuildBPropCutFuncGraph()
|
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 393 def del_cell(self, cell_id=""): argument
394 self._executor.clear_cell(cell_id)
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_hook.py | 64 def cell_hook_function_print_grad(cell_id, grad_input, grad_output): argument
|
| D | test_pynative_hook_grad.py | 40 def ms_record_hook(self, cell_id, grad_input, grad_output): argument
46 def ms_change_grad_double_hook(self, cell_id, grad_input, grad_output): argument
|
| /third_party/gstreamer/gstplugins_bad/gst-libs/gst/mpegts/ |
| D | gst-dvb-descriptor.h | 1022 guint16 cell_id; member
|
| D | gst-dvb-descriptor.c | 2413 cell->cell_id = GST_READ_UINT16_BE (data); in gst_mpegts_descriptor_parse_dvb_t2_delivery_system()
|