Searched refs:kPrimListGetItem (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 77 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
80 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
86 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
89 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
92 …{prim::kPrimTupleGetItem, prim::kPrimTupleSetItem, prim::kPrimListGetItem, prim::kPrimListSetItem}… in OptimizeIRPassLib()
|
| D | clean.cc | 486 } else if (IsPrimitiveCNode(node, prim::kPrimListGetItem)) { in CleanAfterOptA()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | item_tuple_or_list_eliminate.h | 46 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsVNode})(node); in operator()
104 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsVNode})(node); in operator()
152 AnfVisitor::Match(prim::kPrimListGetItem, {IsVNode, IsVNode})(node); in operator()
300 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsVNode})(node); in operator()
370 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | list_append_operation.cc | 46 …elems.push_back(ret->NewCNode({NewValueNode(prim::kPrimListGetItem), arg0_node, NewValueNode(SizeT… in GenerateFuncGraph()
|
| D | unpack_call.cc | 74 …ret_graph->NewCNode({NewValueNode(prim::kPrimListGetItem), para_list, NewValueNode(SizeToLong(i))}… in GenerateFuncGraph()
|
| D | composite.cc | 156 … {NewValueNode(prim::kPrimListGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMake()
432 op = prim::kPrimListGetItem; in GenerateSequeueFuncGraph()
550 …grads.push_back(b->NewCNodeInOrder({NewValueNode(prim::kPrimListGetItem), dout, NewValueNode(i)})); in GenerateFuncGraph()
|
| D | map.cc | 114 … {NewValueNode(prim::kPrimListGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMakeList()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 131 … {"__getitem__", prim::kPrimListGetItem}, // P.list_getitem, in GetMethodMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kpynative.cc | 449 IsPrimitiveCNode(output_node, prim::kPrimListGetItem)) { in UpdateOutputNodeOfTopCell()
588 …rimitiveCNode(cnode, prim::kPrimTupleGetItem) || IsPrimitiveCNode(cnode, prim::kPrimListGetItem)) { in ForgeCNodeAdjoint()
954 getitem_prim = prim::kPrimListGetItem; in BuildMakeSequenceBprop()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 144 {prim::kPrimListGetItem, {InferImplListGetItem, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 197 ASSERT_EQ(prim->name(), kPrimListGetItem->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 603 inline const PrimitivePtr kPrimListGetItem = std::make_shared<Primitive>("list_getitem"); variable
|