Searched refs:kPrimEnvSetItem (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | env_item_eliminate.h | 60 while (IsPrimitiveCNode(env, prim::kPrimEnvSetItem)) { in operator()
124 while (IsPrimitiveCNode(env, prim::kPrimEnvSetItem)) { in operator()
219 if (!IsPrimitiveCNode(node, prim::kPrimEnvSetItem)) { in operator()
253 while (IsPrimitiveCNode(env, prim::kPrimEnvSetItem)) { in operator()
|
| D | branch_culling.cc | 62 … {prim::kPrimEnvSetItem, {1}}, in InConvertWhiteList()
86 {prim::kPrimEnvGetItem, {1}}, {prim::kPrimEnvSetItem, {1}}, in InConvertWhiteList()
|
| D | incorporate_getitem.h | 841 …return IsPrimitive(node.first, prim::kPrimEnvSetItem) || IsPrimitive(node.first, prim::kPrimEnvGet… in ExistEnvNode()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | prim_test.cc | 381 FuncGraphPtr func_graph = MakeFuncGraph(prim::kPrimEnvSetItem, 3); in TEST_F()
401 FuncGraphPtr graph_setitem = MakeFuncGraph(prim::kPrimEnvSetItem, 3); in TEST_F()
431 FuncGraphPtr graph_setitem = MakeFuncGraph(prim::kPrimEnvSetItem, 3); in TEST_F()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 175 {prim::kPrimEnvSetItem, {InferImplEnvSetItem, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | dfunctor.cc | 385 NewValueNode(prim::kPrimEnvSetItem), in AttachFvDoutToTape()
412 NewValueNode(prim::kPrimEnvSetItem), in AttachIndirectFvDoutToTape()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 351 ASSERT_EQ(prim->name(), kPrimEnvSetItem->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 613 inline const PrimitivePtr kPrimEnvSetItem = std::make_shared<Primitive>("env_setitem"); variable
|