Home
last modified time | relevance | path

Searched defs:op_info_ptr (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_metadata.cc31 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAKG); in AkgMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dsingle_tbe_json_creator.cc75 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenDataJson() local
104 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenInputsJson() local
241 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenOutputsJson() local
370 …ctTbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing()
Dtbe_json_creator.cc349 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in GenAttrsDescJson() local
373 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in GenComputeCommonJson() local
556 …l TbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing()
Dfusion_tbe_json_creator.cc303 …ldTbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc35 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in AicpuMetadataInfo() local
Daicpu_kernel_build.cc155 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in SetNodeAttr() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc81 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kAKG); in SupportedTypeList() local
121 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kAKG); in SelectAkgKernel() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc128 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(kernel_name, kernel::OpImplyType::kCPU); in CPUKernelAttrCheck() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc410 const std::shared_ptr<const OpInfo> &op_info_ptr) { in SetKernelBuildInfo()
429 bool ParseMetadata(const CNodePtr &kernel_node, const std::shared_ptr<const OpInfo> &op_info_ptr, P… in ParseMetadata()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc321 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kCPU); in SetKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/
Dop_tiling_adapter.cc234 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name_, node); in InitOpIoName() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc63 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(node_name_, cnode_ptr_); in TbeMetadataInfoEx() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc193 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenTbeSingleKernelJson() local
1053 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(origin_type, cnode); in GenFusionComputeCommonJson() local