Home
last modified time | relevance | path

Searched refs:kLambApplyWeightAssign (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc62 {kAscendDevice, OpLevel_0, prim::kLambApplyWeightAssign}, in GetExpandOps()
205 {prim::kLambApplyWeightAssign, std::make_shared<OpUMonadExpander>(kLambWeightInputIdx)}, in GetExpander()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h398 inline const PrimitivePtr kLambApplyWeightAssign = std::make_shared<Primitive>("LambApplyWeightAssi… variable