Home
last modified time | relevance | path

Searched defs:prim_node (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/mindapi/
Dmindapi_test.cc58 auto prim_node = MakeShared<ValueNode>(prim); in TEST_F() local
185 auto prim_node = MakeShared<ValueNode>(prim); in TEST_F() local
220 auto prim_node = MakeShared<ValueNode>(prim); in TEST_F() local
270 auto prim_node = MakeShared<ValueNode>(prim); in TEST_F() local
432 auto prim_node = MakeShared<ValueNode>(prim); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dtaylor_eliminate.cc89 auto prim_node = GetValueNode<PrimitivePtr>(node); in GetTaylorPrimitive() local
111 auto prim_node = GetValueNode<PrimitivePtr>(node); in TaylorFunctor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/format/
Dto_nchw_format.cc26 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
Dto_nhwc_format.cc45 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Dparallel_node_check.cc106 const auto &prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelConsiderCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/
Dunify_format.cc33 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
84 auto prim_node = cnode->input(0); in InsertPreTransNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dadd_akg_kernel_attrs.cc35 auto prim_node = NewValueNode(common::AnfAlgo::GetCNodePrimitive(cnode)->Clone()); in ClonePrimitive() local
Dreplace_node_by_proxy.cc68 const auto prim_node = NewValueNode(prim); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dgroup_depthwise_op_convert_pass.cc50 auto prim_node = conv_cnode->input(0); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dformat_utils.cc336 auto prim_node = cnode->input(0); in GetFormatSensitiveOpInsertIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dpattern_engine.cc191 auto prim_node = utils::cast<AnfNodePtr>(values_pattern[0]); in UpdateEquivMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/dynamic_obfuscation/
Ddynamic_obfuscation.cc515 mindspore::PrimitivePtr prim_node = one_input_prim_[0]; in get_random_prim() local
700 mindspore::PrimitivePtr prim_node = GetCNodePrimitive(node); in CloneSubGraph() local
772 mindspore::PrimitivePtr prim_node = get_random_prim(obf_type, node); in BuildFakeGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/
Dunify_format.cc143 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dop_adapter_util.cc480 AnfNodePtr prim_node = nullptr; in GetOpIOFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc411 auto prim_node = cnode->input(0); in NodeHasLabel() local
1239 auto prim_node = cnode->input(0); in GetNodeLabel() local
1789 auto prim_node = cnode->input(0); in GetSplitLabel() local
Dfold_pipeline_split_utils.cc158 for (auto &prim_node : END_NODE_BLACK_LIST) { in IsInEndNodeBlackListOrParallelBlackList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/
De2e_dump.cc261 auto prim_node = GetCNodePrimitive(node); in DumpArgsSingleNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.cc482 auto prim_node = raw_cnode->input(0)->cast<ValueNodePtr>(); in UnifyAssembledMappingWithSqueezedFromShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dnode.cc62 auto prim_node = mindspore::NewValueNode(prim); in MSNewOp() local
1270 auto prim_node = node_input->cast<ValueNodePtr>(); in GetOpPrim() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dspecial_func_infer.cc590 ValueNode *prim_node = sub_graph->NewValueNode(AObject::Convert(prim_cast), LOAD_CONST, -1, {}); in InferTensorAsType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph.cc137 auto prim_node = NewValueNode(p->Clone()); in SetInternalOutputAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Danf.cc39 std::pair<bool, PrimitivePtr> IsNeedCheckPrimitiveNode(const AnfNodePtr &prim_node) { in IsNeedCheckPrimitiveNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc287 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelCareNode() local
1136 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsAutoParallelCareNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Daction.cc1773 auto prim_node = cnode->input(0); in KeepValueNodeDuplication() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dpynative_utils.cc2242 auto prim_node = in BuildSpecialNode() local

12