| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_cpu_dynamic_kernel.cc | 49 auto cnode = cnode_ptr_.lock(); in Execute() local
62 auto cnode = cnode_ptr_.lock(); in Initialize() local
113 auto cnode = cnode_ptr_.lock(); in UpdateInputOutputAddr() local
147 auto cnode = cnode_ptr_.lock(); in UpdateExtInfo() local
184 auto cnode = cnode_ptr_.lock(); in UpdateOutputShapeFromExtInfo() local
215 auto cnode = cnode_ptr_.lock(); in PostExecute() local
|
| D | ai_core_dynamic_kernel.cc | 46 auto cnode = cnode_ptr_.lock(); in Execute() local
69 auto cnode = cnode_ptr_.lock(); in ParseCompileJson() local
100 auto cnode = cnode_ptr_.lock(); in UpdateArgs() local
133 auto cnode = cnode_ptr_.lock(); in ComputeTiling() local
150 auto cnode = cnode_ptr_.lock(); in AllocateWorkspace() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | unify_format.cc | 36 STATUS DecideMINDIRConvWeightSrcFormat(const CNodePtr &cnode, schema::Format *src_format) { in DecideMINDIRConvWeightSrcFormat()
56 STATUS DecideTFConvWeightSrcFormat(const CNodePtr &cnode, schema::Format *src_format) { in DecideTFConvWeightSrcFormat()
79 STATUS DecideTFLITEConvWeightSrcFormat(const CNodePtr &cnode, schema::Format *src_format) { in DecideTFLITEConvWeightSrcFormat()
102 STATUS DecideCAFFEConvWeightSrcFormat(const CNodePtr &cnode, schema::Format *src_format) { in DecideCAFFEConvWeightSrcFormat()
108 STATUS DecideONNXConvWeightSrcFormat(const CNodePtr &cnode, schema::Format *src_format) { in DecideONNXConvWeightSrcFormat()
135 STATUS UnifyFormatToNHWC::GetTransNodeFormatType(const CNodePtr &cnode, opt::TransTypePair *trans_i… in GetTransNodeFormatType()
200 STATUS UnifyFormatToNHWC::DecideConvWeightSrcAndDstFormat(const CNodePtr &cnode, schema::Format *sr… in DecideConvWeightSrcAndDstFormat()
224 …rmatToNHWC::ConvertOnnxResizeForConstShape(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ConvertOnnxResizeForConstShape()
268 …tToNHWC::ConvertOnnxResizeForVariableShape(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ConvertOnnxResizeForVariableShape()
298 STATUS UnifyFormatToNHWC::ResizeNodeProcess(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ResizeNodeProcess()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | primitive_adjust.cc | 209 int MoveAttrMapCommon(const CNodePtr &cnode) { in MoveAttrMapCommon()
225 int MoveAttrMapActivation(const CNodePtr &cnode) { in MoveAttrMapActivation()
247 int MoveAttrMapActivationGrad(const CNodePtr &cnode) { in MoveAttrMapActivationGrad()
269 int MoveAttrMapReduce(const CNodePtr &cnode) { in MoveAttrMapReduce()
292 int MoveAttrMapConv2D(const CNodePtr &cnode) { in MoveAttrMapConv2D()
340 int MoveAttrPool(const CNodePtr &cnode) { in MoveAttrPool()
376 int MoveAttrPoolGrad(const CNodePtr &cnode) { in MoveAttrPoolGrad()
412 int MoveAttrMapAdder(const CNodePtr &cnode) { in MoveAttrMapAdder()
443 int MoveAttrMapLayerNorm(const CNodePtr &cnode) { in MoveAttrMapLayerNorm()
463 int MoveAttrMapResize(const CNodePtr &cnode) { in MoveAttrMapResize()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | redundant_op_remove_pass.cc | 31 int ProcessInputIsMonad(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ProcessInputIsMonad()
76 int ProcessDependencyWithTwoNodes(const FuncGraphPtr &func_graph, const CNodePtr &cnode, bool pre_n… in ProcessDependencyWithTwoNodes()
115 int ProcessInputHaveDependency(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ProcessInputHaveDependency()
143 auto cnode = anf_node->cast<CNodePtr>(); in ReplaceOp() local
180 auto cnode = anf_node->cast<CNodePtr>(); in ReplaceUpdateStateOp() local
197 auto cnode = anf_node->cast<CNodePtr>(); in ReplaceTupleGetItem() local
234 auto cnode = anf_node->cast<CNodePtr>(); in RemoveDropoutOp() local
267 int RemoveRedundantOpPass::GetConstDataFromInputNode(const CNodePtr &cnode, lite::DataInfo *data_in… in GetConstDataFromInputNode()
293 auto cnode = anf_node->cast<CNodePtr>(); in RemoveInvalidPadOp() local
343 auto cnode = anf_node->cast<CNodePtr>(); in RemoveInvalidTransposeOp() local
|
| D | transpose_strategy.cc | 39 STATUS GetPostNodes(const FuncGraphPtr &func_graph, const CNodePtr &cnode, std::vector<AnfNodePtr> … in GetPostNodes()
59 bool JudgeIs4DInput(NodeInferShape *node_infer_shape, const CNodePtr &cnode) { in JudgeIs4DInput()
93 int TransformAttrByAxes(const FuncGraphPtr &func_graph, const CNodePtr &cnode, size_t input_index, in TransformAttrByAxes()
127 STATUS ChangeCommonOp(const FuncGraphPtr &func_graph, const CNodePtr &cnode, FormatTransNodeType tr… in ChangeCommonOp()
152 STATUS ChangeOpCrop(const FuncGraphPtr &func_graph, const CNodePtr &cnode, FormatTransNodeType tran… in ChangeOpCrop()
200 STATUS ChangeOpPad(const FuncGraphPtr &func_graph, const CNodePtr &cnode, FormatTransNodeType trans… in ChangeOpPad()
261 STATUS ChangeOpSlice(const FuncGraphPtr &func_graph, const CNodePtr &cnode, FormatTransNodeType tra… in ChangeOpSlice()
302 STATUS ChangeOpStrideSlice(const FuncGraphPtr &func_graph, const CNodePtr &cnode, FormatTransNodeTy… in ChangeOpStrideSlice()
342 …ansposeStrategy::TransposePairFuseWhenInsert(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in TransposePairFuseWhenInsert()
366 …tr TransposeStrategy::TransposeDependOnShape(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in TransposeDependOnShape()
[all …]
|
| D | decrease_transpose_algo.cc | 105 for (auto &cnode : cnodes) { in SetTransType() local
149 int ConvertTensorToNCOrNH(const FuncGraphPtr &func_graph, const CNodePtr &cnode, size_t index, FmkT… in ConvertTensorToNCOrNH()
205 …DecreaseTransposeAlgo::PostTransposeFusion(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in PostTransposeFusion()
234 STATUS DecreaseTransposeAlgo::GenNewInput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in GenNewInput()
275 …US DecreaseTransposeAlgo::InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in InsertPreTransNode()
311 …US DecreaseTransposeAlgo::InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in InsertPreTransNode()
366 …S DecreaseTransposeAlgo::InsertPostTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in InsertPostTransNode()
407 … DecreaseTransposeAlgo::HandleGraphMultiNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in HandleGraphMultiNode()
476 int DecreaseTransposeAlgo::SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphInput()
580 int DecreaseTransposeAlgo::SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph… in SetSubGraphAbstract()
[all …]
|
| D | node_infershape.cc | 71 bool NodeInferShape::JudgeOpSupportInfer(const CNodePtr &cnode) { in JudgeOpSupportInfer()
89 STATUS NodeInferShape::InferShape(const CNodePtr &cnode) { in InferShape()
176 std::vector<int> NodeInferShape::GetInputShape(const CNodePtr &cnode, size_t index) { in GetInputShape()
207 std::vector<int> NodeInferShape::GetIntVecInput(const CNodePtr &cnode, size_t index) { in GetIntVecInput()
241 STATUS NodeInferShape::GetCNodeInputTensors(const CNodePtr &cnode, std::vector<lite::Tensor *> *inp… in GetCNodeInputTensors()
291 STATUS NodeInferShape::GetCNodeConstInput(const CNodePtr &cnode, std::vector<lite::Tensor *> *const… in GetCNodeConstInput()
317 STATUS NodeInferShape::GetCNodeVarInput(const CNodePtr &cnode, std::vector<lite::Tensor *> *var_ms_… in GetCNodeVarInput()
375 STATUS NodeInferShape::GetCNodeOutputTensors(const CNodePtr &cnode, std::vector<lite::Tensor *> *ou… in GetCNodeOutputTensors()
475 STATUS NodeInferShape::SetCNodeAbstract(const std::shared_ptr<CNode> &cnode, const std::vector<lite… in SetCNodeAbstract()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | auto_monad.cc | 111 RefInputs GetRefInputs(const CNodePtr &cnode) { in GetRefInputs()
124 bool HasRefInput(const CNodePtr &cnode) { in HasRefInput()
152 PrimitivePtr GetPrimitive(const CNodePtr &cnode) { in GetPrimitive()
160 FuncGraphPtr GetFuncGraph(const CNodePtr &cnode) { in GetFuncGraph()
168 ClassTypePtr GetClassType(const CNodePtr &cnode) { in GetClassType()
181 CNodePtr GetFuncCNode(const CNodePtr &cnode) { in GetFuncCNode()
190 ParameterPtr GetFuncParameter(const CNodePtr &cnode) { in GetFuncParameter()
199 prim::MultitypeFuncGraphPtr GetFuncMultitypeFuncGraph(const CNodePtr &cnode) { in GetFuncMultitypeFuncGraph()
345 for (auto &cnode : order_list) { in UpdateOrderList() local
351 …static void PushToOrderList(const FuncGraphPtr &fg, const CNodePtr &cnode, OrderedSet<CNodePtr> *n… in PushToOrderList()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 88 STATUS WeightQuantizer::DoConvQuantize(const CNodePtr &cnode) { in DoConvQuantize()
139 STATUS WeightQuantizer::DoMulQuantize(const CNodePtr &cnode) { in DoMulQuantize()
190 STATUS WeightQuantizer::DoLstmQuantize(const CNodePtr &cnode) { in DoLstmQuantize()
223 STATUS WeightQuantizer::DoGatherQuantize(const CNodePtr &cnode) { in DoGatherQuantize()
267 STATUS WeightQuantizer::DoOptimizerQuantize(const CNodePtr &cnode) { in DoOptimizerQuantize()
315 STATUS WeightQuantizer::DoMarkWeightQuantizeIfQuantized(const CNodePtr &cnode) { in DoMarkWeightQuantizeIfQuantized()
345 STATUS WeightQuantizer::ProcessLstmWeightByIndex(const CNodePtr &cnode, const PrimitivePtr &primiti… in ProcessLstmWeightByIndex()
395 for (auto &cnode : func_graph->GetOrderedCnodes()) { in MarkWeightQuantizationInNodes() local
414 for (auto &cnode : func_graph->GetOrderedCnodes()) { in DoQuantize() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | merge_addn.h | 49 auto cnode = node->cast<CNodePtr>(); in operator() local
60 void Visit(const CNodePtr &cnode) override { in Visit()
172 auto cnode = node->cast<CNodePtr>(); in operator() local
179 void Visit(const CNodePtr &cnode) override { in Visit()
|
| D | switch_or_switch_layer_defer_inline.h | 35 auto cnode = node->cast<CNodePtr>(); in operator() local
52 auto cnode = node->cast<CNodePtr>(); in operator() local
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | parallel_node_check.cc | 55 bool IsInAllGatherNodeList(const CNodePtr &cnode) { in IsInAllGatherNodeList()
64 bool IsInTrivialNodeList(const CNodePtr &cnode) { in IsInTrivialNodeList()
73 bool IsParallelConsiderCNode(const CNodePtr &cnode) { in IsParallelConsiderCNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | optimize_dependence.cc | 34 CNodePtr CreateNewDependNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in CreateNewDependNode()
52 CNodePtr CheckIsolatedVirtualNode(const CNodePtr &cnode) { in CheckIsolatedVirtualNode()
71 AnfNodePtr EliminateIsolatedVirtualNodeInput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in EliminateIsolatedVirtualNodeInput()
90 auto cnode = node->cast<CNodePtr>(); in GetReplaceNode() local
114 AnfNodePtr ReplaceMakeTuple(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in ReplaceMakeTuple()
159 std::vector<size_t> SearchTransDataAndCast(const CNodePtr &cnode) { in SearchTransDataAndCast()
181 auto cnode = dyn_cast<CNode>(node); in Process() local
|
| D | add_training_attr.cc | 37 bool CheckOP(const FuncGraphManagerPtr &manager, const AnfNodePtr &cnode, const std::unordered_set<… in CheckOP()
54 void AddAttrTraining(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in AddAttrTraining()
85 auto cnode = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 33 auto cnode = cnode_ptr_.lock(); in Initialize() local
59 auto cnode = cnode_ptr_.lock(); in RebuildDependTensor() local
82 auto cnode = cnode_ptr_.lock(); in InferShape() local
96 auto cnode = cnode_ptr_.lock(); in InferShapeRecursive() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | deal_ref_and_split_unsupported_transdata.cc | 35 auto cnode = cur_node->cast<CNodePtr>(); in FindRefOriginNode() local
65 … const CNodePtr &cnode, const size_t output_index, in AddRefNodePairToKernelGraph()
75 …nSupportedTransdata::AddRefPairToKernelGraph(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in AddRefPairToKernelGraph()
100 … const CNodePtr &cnode, size_t output_index, in AddAdditionalToRefOutput()
155 const CNodePtr &cnode, in MakeDependency()
170 auto cnode = orig_cnode; in DealRefForMultipleOutput() local
207 …const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::shared_ptr<kernel::OpInfo> &op_i… in DealRefSingleOutput() argument
248 auto cnode = node->cast<CNodePtr>(); in Process() local
|
| D | set_fracz_group_attr.cc | 37 AnfNodePtr GetOutputItem(const FuncGraphManagerPtr &manager, const CNodePtr &cnode, int64_t groups, in GetOutputItem()
74 auto cnode = node->cast<CNodePtr>(); in HasFraczGroupAttrAndSet() local
105 …nelWithIndex> GetCNodeNeighborFraczNodes(const FuncGraphManagerPtr &manager, const CNodePtr &cnode, in GetCNodeNeighborFraczNodes()
163 auto cnode = node->cast<CNodePtr>(); in GetNeighborFraczNodes() local
182 bool SetAttrFraczGroup(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in SetAttrFraczGroup()
213 auto cnode = node->cast<CNodePtr>(); in Run() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.cc | 29 STATUS ToFormatBase::GenNewInput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::… in GenNewInput()
68 STATUS ToFormatBase::ModifyCNode(const CNodePtr &cnode) { in ModifyCNode()
109 STATUS ToFormatBase::InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in InsertPreTransNode()
139 STATUS ToFormatBase::InsertPostTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, in InsertPostTransNode()
257 STATUS ToFormatBase::HandleGraphNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { in HandleGraphNode()
297 auto cnode = node->cast<CNodePtr>(); in BasicProcess() local
346 auto cnode = node->cast<CNodePtr>(); in ConvWeightFormatTrans() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_inputs_adjust.cc | 40 bool CheckResize(const CNodePtr &cnode) { in CheckResize()
54 lite::STATUS ReorderCnodeInputs(CNode *cnode, const std::vector<size_t> &perm) { in ReorderCnodeInputs()
137 auto cnode = node->cast<CNodePtr>(); in AdjustSlice() local
164 auto cnode = node->cast<CNodePtr>(); in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 39 static bool IsInEndNodeBlackList(const CNodePtr &cnode) { in IsInEndNodeBlackList()
56 AnfNodePtr FindAccuGrad(const CNodePtr &cnode) { in FindAccuGrad()
90 auto cnode = node->cast<CNodePtr>(); in SetStridedSliceStrategy() local
113 auto cnode = node_user.first->cast<CNodePtr>(); in InsertVirtualAssignAdd() local
137 auto cnode = recv->cast<CNodePtr>(); in InsertVirtualAccuGrad() local
171 auto cnode = node->cast<CNodePtr>(); in HandleReceiveParam() local
368 auto cnode = node->cast<CNodePtr>(); in GetMicroBatch() local
431 auto cnode = node_user.first->cast<CNodePtr>(); in BroadCastNeedGrad() local
462 auto cnode = node->cast<CNodePtr>(); in GetPreNode() local
491 auto cnode = node->cast<CNodePtr>(); in LastStageEndNode() local
[all …]
|
| /third_party/mindspore/mindspore/lite/examples/converter_extend/src/ |
| D | pass_registry_tutorial.cc | 34 auto cnode = node->cast<CNodePtr>(); in CheckPrimitiveTypeTutorial() local
44 AnfNodePtr PassTutorial::CreateCustomOp(const api::FuncGraphPtr func_graph, const CNodePtr &cnode) { in CreateCustomOp()
87 auto cnode = node->cast<CNodePtr>(); in Execute() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | eltwise_mapper.cc | 23 STATUS EltWiseMapper::Mapper(const CNodePtr &cnode) { in Mapper()
31 STATUS EltWiseMapper::AddAttrForDynInputPrimitive(const CNodePtr &cnode) { in AddAttrForDynInputPrimitive()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/common/ |
| D | utils.cc | 54 auto cnode = node->cast<mindspore::CNodePtr>(); in CheckPrimitiveType() local
62 STATUS GetShapeVectorFromCNode(const mindspore::CNodePtr &cnode, std::vector<int64_t> *shape_vector… in GetShapeVectorFromCNode()
109 auto cnode = node->cast<CNodePtr>(); in GetTypeFromNode() local
171 std::string GetCNodeTargetFuncName(const CNodePtr &cnode) { in GetCNodeTargetFuncName()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_property_checker.cc | 35 static bool CheckStridedSlice(const CNodePtr &cnode) { in CheckStridedSlice()
77 static bool CheckTopK(const CNodePtr &cnode) { in CheckTopK()
85 bool TbePropertyChecker::CheckTbeProperties(const mindspore::CNodePtr &cnode) { in CheckTbeProperties()
|