Searched refs:kPrimAssignAdd (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 54 {kAllTarget, OpLevel_0, prim::kPrimAssignAdd}, in GetExpandOps()
202 {prim::kPrimAssignAdd, std::make_shared<OpUMonadExpander>(kAssignInputIdx)}, in GetExpander()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_cpu_kernel.cc | 395 if (kernel_name_ == prim::kPrimAssignAdd->name()) { in InitComputeFunc()
464 if (kernel_name_ == prim::kPrimAssignAdd->name()) { in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 72 {prim::kPrimAssignAdd, {1}}, in InConvertWhiteList()
91 {prim::kPrimAssign, {1}}, {prim::kPrimAssignAdd, {1}}, in InConvertWhiteList()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 429 auto prim = prim::kPrimAssignAdd; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 540 inline const PrimitivePtr kPrimAssignAdd = std::make_shared<Primitive>("AssignAdd"); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 38 …et<std::string> kOpAssignKernelNameList = {prim::kPrimAssign->name(), prim::kPrimAssignAdd->name(),
|