Home
last modified time | relevance | path

Searched refs:kAKG (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Doplib.cc73 case kAKG: in ImplTypeToStr()
91 OpImplyType imply_type = kAKG; in RegOp()
212 } else if (imply_type == kAKG) { in DecodeOpInfo()
372 if (imply_type == kAKG && op_info->processor() != target_processor) { in FindOp()
Dopinfo.h28 enum OpImplyType { kAKG = 0, kTBE = 1, kAICPU = 2, kCPU }; enumerator
/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()
Dakg_kernel_json_generator.cc53 op_info->set_imply_type(OpImplyType::kAKG); in Run()
454 return mindspore::kernel::OpLib::FindOp(AnfAlgo::GetCNodeName(anf_node), OpImplyType::kAKG); in ExtractOpInfo()
/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()
121 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kAKG); in SelectAkgKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc420 if (imply_type == kAKG) { in SetKernelBuildInfo()