Searched defs:cell_id (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/tests/st/hook/ |
| D | test_hook_function.py | 93 def cell_hook_function(cell_id, grad_input, grad_output): argument
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 2019 void GradExecutor::PushCellStack(const std::string &cell_id) { cell_stack_.push(cell_id); } in PushCellStack()
2047 auto cell_id = GetId(cell); in GetCellId() local
2100 bool GradExecutor::IsBpropGraph(const std::string &cell_id) { in IsBpropGraph()
2114 void GradExecutor::ClearCellRes(const std::string &cell_id) { in ClearCellRes()
2182 void GradExecutor::InitResourceAndDfBuilder(const std::string &cell_id, const py::args &args) { in InitResourceAndDfBuilder()
2224 const auto &cell_id = GetCellId(cell, args); in NewGraphInner() local
2265 void GradExecutor::MakeNewTopGraph(const string &cell_id, const py::args &args, bool is_topest) { in MakeNewTopGraph()
2382 const auto &cell_id = GetCellId(cell, args); in EndGraphInner() local
2471 const auto &cell_id = GetCellId(cell, args); in DoGradForCustomBprop() local
2502 std::string GradExecutor::GetAlreadyRunCellId(const std::string &cell_id) { in GetAlreadyRunCellId()
[all …]
|
| D | pynative_execute.h | 51 std::string cell_id; member
92 const std::string &cell_id() const { return cell_id_; } in cell_id() function
|
| /third_party/mindspore/tests/st/pynative/ |
| 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
|
| D | test_pynative_hook.py | 64 def cell_hook_function_print_grad(cell_id, grad_input, grad_output): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | debug_ops.py | 356 def __init__(self, hook_fn, cell_id=""): argument
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_hook.py | 54 def cell_hook_function(cell_id, grad_input, grad_output): argument
|
| /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
|
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 393 def del_cell(self, cell_id=""): argument
|
| /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
|
| D | kpynative.cc | 905 auto cell_id = GetValue<std::string>(prim->GetAttr("cell_id")); in BuildBPropCutFuncGraph() local
|
| /third_party/gstreamer/gstplugins_bad/gst-libs/gst/mpegts/ |
| D | gst-dvb-descriptor.h | 1022 guint16 cell_id; member
|