| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 32 void SetAkgAttrsForFour2Five(const AnfNodePtr &anf_node) { in SetAkgAttrsForFour2Five()
50 void SetAkgAttrsForFive2Four(const AnfNodePtr &anf_node) { in SetAkgAttrsForFive2Four()
76 void SetAkgAttrsForCast(const AnfNodePtr &anf_node) { in SetAkgAttrsForCast()
90 void SetAkgAttrsForBNGrad1(const AnfNodePtr &anf_node) { in SetAkgAttrsForBNGrad1()
98 void SetAkgAttrsForBNGrad2(const AnfNodePtr &anf_node) { in SetAkgAttrsForBNGrad2()
122 void SetAkgAttrsForBNGrad3(const AnfNodePtr &anf_node) { in SetAkgAttrsForBNGrad3()
130 void SetAkgAttrsForFusedBN1(const AnfNodePtr &anf_node) { in SetAkgAttrsForFusedBN1()
139 void SetAkgAttrsForFusedBN2(const AnfNodePtr &anf_node) { in SetAkgAttrsForFusedBN2()
148 void SetAkgAttrsForFusedBN3(const AnfNodePtr &anf_node) { in SetAkgAttrsForFusedBN3()
157 void SetAkgAttrsForConvBN1(const AnfNodePtr &anf_node) { in SetAkgAttrsForConvBN1()
[all …]
|
| D | akg_kernel_json_generator.cc | 30 std::vector<int> GetDynInputSize(const AnfNodePtr &anf_node) { in GetDynInputSize()
47 OpInfoPtr Run(const AnfNodePtr &anf_node) { in Run()
143 TypeId AkgKernelJsonGenerator::GetInputDataType(const AnfNodePtr &anf_node, size_t real_index) cons… in GetInputDataType() argument
148 std::vector<size_t> AkgKernelJsonGenerator::GetInputShape(const AnfNodePtr &anf_node, size_t real_i… in GetInputShape() argument
153 std::string AkgKernelJsonGenerator::GetInputFormat(const AnfNodePtr &anf_node, size_t real_index) c… in GetInputFormat() argument
157 TypeId AkgKernelJsonGenerator::GetOutputDataType(const AnfNodePtr &anf_node, size_t index) const { in GetOutputDataType() argument
162 std::vector<size_t> AkgKernelJsonGenerator::GetOutputShape(const AnfNodePtr &anf_node, size_t index… in GetOutputShape() argument
167 std::string AkgKernelJsonGenerator::GetOutputFormat(const AnfNodePtr &anf_node, size_t index) const… in GetOutputFormat() argument
171 bool AkgKernelJsonGenerator::CreateInputDescJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_in… in CreateInputDescJson()
224 bool AkgKernelJsonGenerator::CreateOutputDescJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_i… in CreateOutputDescJson()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 26 bool TbeJsonUtils::GetInputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &inp… in GetInputsRealNum()
53 bool TbeJsonUtils::GetOutputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &ou… in GetOutputsRealNum()
73 bool TbeJsonUtils::IsNeedChangeDefaultFormat(const AnfNodePtr &anf_node) { in IsNeedChangeDefaultFormat()
79 std::vector<int64_t> TbeJsonUtils::GetInputOriShapeForTbeBuild(const AnfNodePtr &anf_node, size_t r… in GetInputOriShapeForTbeBuild()
85 std::vector<int64_t> TbeJsonUtils::GetInputDeviceShapeForTbeBuild(const AnfNodePtr &anf_node, size_… in GetInputDeviceShapeForTbeBuild()
97 std::vector<int64_t> TbeJsonUtils::GetOutputOriShapeForTbeBuild(const AnfNodePtr &anf_node, size_t … in GetOutputOriShapeForTbeBuild()
113 std::vector<int64_t> TbeJsonUtils::GetOutputDeviceShapeForTbeBuild(const AnfNodePtr &anf_node, size… in GetOutputDeviceShapeForTbeBuild()
|
| D | single_tbe_json_creator.cc | 31 bool SingleTbeJsonCreator::GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { in GenJson()
54 bool SingleTbeJsonCreator::GenOpListJson(const AnfNodePtr &anf_node, std::vector<nlohmann::json> *o… in GenOpListJson()
69 void SingleTbeJsonCreator::GenDataJson(const AnfNodePtr &anf_node, const nlohmann::json &compute_js… in GenDataJson()
99 bool SingleTbeJsonCreator::GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenInputsJson()
147 void SingleTbeJsonCreator::GenInputDescJson(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputDescJson()
177 void SingleTbeJsonCreator::GenOutputDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, in GenOutputDescJson()
187 bool SingleTbeJsonCreator::AssignInputsJson(const AnfNodePtr &anf_node, const std::vector<nlohmann:… in AssignInputsJson()
236 bool SingleTbeJsonCreator::GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json)… in GenOutputsJson()
275 bool SingleTbeJsonCreator::AssignOutputsJson(const AnfNodePtr &anf_node, in AssignOutputsJson()
313 void SingleTbeJsonCreator::GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenOtherJson()
[all …]
|
| D | tbe_json_creator.cc | 221 bool TbeJsonCreator::GenComputeJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenComputeJson()
296 void TbeJsonCreator::GenAttrsJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_info, nlohmann::j… in GenAttrsJson()
343 void TbeJsonCreator::GenAttrsDescJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { in GenAttrsDescJson()
367 void TbeJsonCreator::GenComputeCommonJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json)… in GenComputeCommonJson()
393 void TbeJsonCreator::GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_outpu… in GenDescJson()
498 void TbeJsonCreator::GenInputConstValue(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputConstValue()
531 bool TbeJsonCreator::AttrsJsonPreProcessing(const AnfNodePtr &anf_node, std::vector<OpAttrPtr> *att… in AttrsJsonPreProcessing()
536 void TbeJsonCreator::GenOutputDataDescJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json… in GenOutputDataDescJson()
556 bool TbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_p… in AttrsJsonPostProcessing()
|
| D | tbe_json_creator.h | 69 virtual bool GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { return false; } in GenJson()
76 …virtual bool GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { return fals… in GenInputsJson()
77 …virtual bool GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) { return fal… in GenOutputsJson()
80 virtual void GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_json) {} in GenOtherJson()
|
| D | fusion_tbe_json_creator.cc | 170 bool FusionBuildTbeJsonCreator::GenInputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_j… in GenInputsJson()
242 bool FusionBuildTbeJsonCreator::GenOutputsJson(const AnfNodePtr &anf_node, nlohmann::json *compute_… in GenOutputsJson()
281 void FusionBuildTbeJsonCreator::GenReusedOutputDesc(const AnfNodePtr &anf_node, size_t index, size_… in GenReusedOutputDesc()
303 bool FusionBuildTbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr… in AttrsJsonPostProcessing()
310 void FusionBuildTbeJsonCreator::GenOtherJson(const AnfNodePtr &anf_node, nlohmann::json *compute_js… in GenOtherJson()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 33 bool HcomUtil::GetKernelInputShape(const AnfNodePtr &anf_node, vector<vector<size_t>> *hccl_kernel_… in GetKernelInputShape()
45 bool HcomUtil::GetKernelOutputShape(const AnfNodePtr &anf_node, vector<vector<size_t>> *hccl_kernel… in GetKernelOutputShape()
65 bool HcomUtil::GetHcomDataType(const AnfNodePtr &anf_node, vector<HcclDataType> *data_type_list) { in GetHcomDataType()
121 bool HcomUtil::GetHcomCount(const AnfNodePtr &anf_node, const vector<HcclDataType> &data_type_list, in GetHcomCount()
187 bool HcomUtil::GetHcomOperationType(const AnfNodePtr &anf_node, HcclReduceOp *op_type) { in GetHcomOperationType()
212 bool HcomUtil::GetHcomRootId(const AnfNodePtr &anf_node, uint32_t *root_id) { in GetHcomRootId()
226 bool HcomUtil::GetHcomSrcRank(const AnfNodePtr &anf_node, uint32_t *src_rank) { in GetHcomSrcRank()
240 bool HcomUtil::GetHcomDestRank(const AnfNodePtr &anf_node, uint32_t *dest_rank) { in GetHcomDestRank()
254 bool HcomUtil::GetHcomReceiveType(const AnfNodePtr &anf_node, TypeId *receive_type) { in GetHcomReceiveType()
268 void HcomUtil::GetHcomGroup(NotNull<const AnfNodePtr &> anf_node, NotNull<std::string *> group) { in GetHcomGroup()
|
| D | hcom_all_to_all.cc | 31 bool HcomAllToAllKernel::Init(const AnfNodePtr &anf_node) { in Init()
66 auto anf_node = anf_node_.lock(); in GenTask() local
|
| D | hccl_kernel.cc | 82 bool HcclKernel::Init(const AnfNodePtr &anf_node) { in Init()
156 auto anf_node = anf_node_.lock(); in GetOutputSizeList() local
212 auto anf_node = anf_node_.lock(); in GenTask() local
|
| D | hccl_kernel_build.cc | 26 KernelModPtr HcclOpBuild(const AnfNodePtr &anf_node) { in HcclOpBuild()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 42 bool SetIOIputSize(const std::shared_ptr<AnfNode> &anf_node, const size_t &input_num, in SetIOIputSize()
77 bool SetIOSize(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &k… in SetIOSize()
144 void SetNodeAttr(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeAttr()
179 void SetNodeInputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeInputs()
219 void SetNodeOutputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodeOutputs()
249 void SetNodedefProto(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef *proto) { in SetNodedefProto()
266 bool CreateNodeDefBytes(const std::shared_ptr<AnfNode> &anf_node, in CreateNodeDefBytes()
295 const std::shared_ptr<AnfNode> &anf_node, size_t input_num) { in SetExtInfoInputShapeType()
335 const std::shared_ptr<AnfNode> &anf_node, size_t output_num) { in SetExtInfoOutputShapeType()
362 void CreateExtInfo(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod… in CreateExtInfo()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 79 bool TbeAdapter::DynamicInputAdjusted(const std::shared_ptr<AnfNode> &anf_node, in DynamicInputAdjusted()
103 void TbeAdapter::InputOrderPass(const std::shared_ptr<AnfNode> &anf_node, in InputOrderPass()
143 void TbeAdapter::FusionInputOrderPass(const std::shared_ptr<AnfNode> &anf_node, in FusionInputOrderPass()
194 bool TbeAdapter::RunAttrPass(const mindspore::AnfNodePtr &anf_node, const std::vector<OpAttrPtr> &o… in RunAttrPass()
206 void TbeAdapter::MaxiOrMinimumGradAttrJsonPass(const AnfNodePtr &anf_node, in MaxiOrMinimumGradAttrJsonPass()
232 void TbeAdapter::CastAttrJsonPass(const mindspore::AnfNodePtr &anf_node, in CastAttrJsonPass()
255 void TbeAdapter::GenTopKV2IndicesTensorInfo(const std::shared_ptr<mindspore::AnfNode> &anf_node, in GenTopKV2IndicesTensorInfo()
434 void TbeAdapter::CastAttrJsonPrePass(const AnfNodePtr &anf_node, std::vector<OpAttrPtr> *op_info_at… in CastAttrJsonPrePass()
461 void TbeAdapter::CastAttrJsonPost(const AnfNodePtr &anf_node, nlohmann::json *attrs_json) { in CastAttrJsonPost()
475 void TbeAdapter::LayerNormAttrJsonPost(const AnfNodePtr &anf_node, nlohmann::json *attrs_json) { in LayerNormAttrJsonPost()
|
| D | tbe_kernel_build.cc | 133 std::vector<int64_t> GetOutputShapeForTbeBuild(const AnfNodePtr &anf_node, size_t real_index) { in GetOutputShapeForTbeBuild()
149 …t64_t> GetOutputDeviceShapeForTbeBuild(const kCreaterType creater_type, const AnfNodePtr &anf_node, in GetOutputDeviceShapeForTbeBuild()
165 std::vector<int64_t> GetInputShapeForTbeBuild(const AnfNodePtr &anf_node, size_t real_index) { in GetInputShapeForTbeBuild()
171 …nt64_t> GetInputDeviceShapeForTbeBuild(const kCreaterType creater_type, const AnfNodePtr &anf_node, in GetInputDeviceShapeForTbeBuild()
188 bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptr<mindspore::AnfNode> &anf_no… in GenTbeSingleKernelJson()
284 void TbeKernelJsonCreator::GenValidInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t r… in GenValidInputDescJson()
316 bool TbeKernelJsonCreator::GenInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t real_i… in GenInputDescJson()
342 bool TbeKernelJsonCreator::GenInputList(const std::shared_ptr<AnfNode> &anf_node, size_t input_tens… in GenInputList()
390 bool GetInputNameAndRealNum(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<OpIOInf… in GetInputNameAndRealNum()
420 bool TbeKernelJsonCreator::GenTbeInputsJson(const std::shared_ptr<AnfNode> &anf_node, in GenTbeInputsJson()
[all …]
|
| D | tbe_adapter.h | 57 …static void InputOrderPass(const std::shared_ptr<AnfNode> &anf_node, std::vector<T> const &inputs_… in InputOrderPass()
97 …static bool DynamicInputAdjusted(const std::shared_ptr<AnfNode> &anf_node, std::vector<T> const &i… in DynamicInputAdjusted()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | redundant_op_remove_pass.cc | 136 int RemoveRedundantOpPass::ReplaceOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &manager… in ReplaceOp()
175 …RedundantOpPass::ReplaceUpdateStateOp(const FuncGraphPtr &func_graph, const AnfNodePtr &anf_node) { in ReplaceUpdateStateOp()
189 int RemoveRedundantOpPass::ReplaceTupleGetItem(const AnfNodePtr &anf_node, const FuncGraphManagerPt… in ReplaceTupleGetItem()
227 int RemoveRedundantOpPass::RemoveDropoutOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr &m… in RemoveDropoutOp()
288 int RemoveRedundantOpPass::RemoveInvalidPadOp(const AnfNodePtr &anf_node, const FuncGraphManagerPtr… in RemoveInvalidPadOp()
342 int RemoveRedundantOpPass::RemoveInvalidTransposeOp(const AnfNodePtr &anf_node, const FuncGraphMana… in RemoveInvalidTransposeOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | shape_ops_splitter.cc | 35 AnfNodePtr CloneCNode(const AnfNodePtr &anf_node) { in CloneCNode()
95 for (const auto &anf_node : todos) { in Process() local
119 for (const auto &anf_node : todos) { in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | memcpy_async.cc | 72 bool MemCpyAsyncKernel::Init(const mindspore::AnfNodePtr &anf_node) { in Init()
79 void MemCpyAsyncKernel::GetInputOutputDataType(const AnfNodePtr &anf_node) { in GetInputOutputDataType()
88 void MemCpyAsyncKernel::GetInputOutputTotalCount(const AnfNodePtr &anf_node) { in GetInputOutputTotalCount()
|
| D | tensor_copy_slices.cc | 71 bool TensorCopySlices::Init(const mindspore::AnfNodePtr &anf_node) { in Init()
87 void TensorCopySlices::GetInputOutputInfo(const AnfNodePtr &anf_node) { in GetInputOutputInfo()
113 void TensorCopySlices::GetInputOutputTotalCount(const AnfNodePtr &anf_node) { in GetInputOutputTotalCount()
|
| D | rt_kernel_build.cc | 26 KernelModPtr RtOpBuild(const AnfNodePtr &anf_node) { in RtOpBuild()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 44 virtual bool CheckAICoreSupported(const AnfNodePtr &anf_node, in CheckAICoreSupported()
48 virtual bool CheckAICPUSupported(const AnfNodePtr &anf_node, in CheckAICPUSupported()
|
| /third_party/mindspore/tests/ut/cpp/stub/aicpu/ |
| D | aicpu_stub.cc | 24 KernelModPtr AicpuOpBuild(const AnfNodePtr &anf_node) { return nullptr; } in AicpuOpBuild()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/ |
| D | profiling_utils.cc | 283 void ProfilingUtils::InsertProfilingTraceFp(const mindspore::AnfNodePtr &anf_node, in InsertProfilingTraceFp()
300 void ProfilingUtils::InsertProfilingTraceJobId(const AnfNodePtr &anf_node, NotNull<session::KernelG… in InsertProfilingTraceJobId()
308 CNodePtr ProfilingUtils::CreateProfilingCNodeWithStream(const mindspore::AnfNodePtr &anf_node, in CreateProfilingCNodeWithStream()
317 void ProfilingUtils::InsertProfilingCustomOp(const AnfNodePtr &anf_node, const ProfilingTraceInfo &… in InsertProfilingCustomOp()
340 void ProfilingUtils::InsertProfilingTraceBpEnd(const AnfNodePtr &anf_node, in InsertProfilingTraceBpEnd()
355 void ProfilingUtils::InsertProfilingTraceIterEnd(const AnfNodePtr &anf_node, in InsertProfilingTraceIterEnd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_build.cc | 26 KernelModPtr HostOpBuild(const std::shared_ptr<AnfNode> &anf_node) { in HostOpBuild()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 46 static kernel::KernelModPtr SerialCompileImpl(const AnfNodePtr &anf_node) { in SerialCompileImpl()
78 for (const auto &anf_node : kernels) { in KernelBuildParallelCompile() local
123 for (const auto &anf_node : other_nodes) { in KernelBuildParallelCompile() local
357 for (const auto &anf_node : kernel_graph->execution_order()) { in TbeClearZeroNodeFusion() local
427 for (const auto &anf_node : kernel_graph->execution_order()) { in KernelBuildPreprocess() local
|