Home
last modified time | relevance | path

Searched refs:kAICPU (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Doplib.cc75 case kAICPU: in ImplTypeToStr()
94 OpImplyType imply_type = kAICPU; in RegOp()
262 if (imply_type != kAICPU) { in DecodeAttr()
360 if (is_gpu && (imply_type == kTBE || imply_type == kAICPU)) { in FindOp()
Dopinfo.h28 enum OpImplyType { kAKG = 0, kTBE = 1, kAICPU = 2, kCPU }; enumerator
/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()
Daicpu_kernel_build.cc155 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in SetNodeAttr()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc422 } else if (imply_type == kAICPU) { in SetKernelBuildInfo()