Searched refs:OpImplyType (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | oplib.h | 34 static std::shared_ptr<OpInfo> FindOp(const std::string &op_name, OpImplyType imply_type,
42 …static bool DecodeOpInfo(const nlohmann::json &obj, OpImplyType imply_type, const std::string &imp…
43 …static bool DecodeAttr(const nlohmann::json &obj, OpImplyType imply_type, const std::shared_ptr<Op…
48 static bool DecodeInputOutput(const nlohmann::json &obj, OpImplyType imply_type, OpIOType io_type,
|
| D | oplib.cc | 69 static std::string ImplTypeToStr(OpImplyType impl_type) { in ImplTypeToStr()
88 OpImplyType imply_type = kTBE; in RegOp()
91 OpImplyType imply_type = kAKG; in RegOp()
94 OpImplyType imply_type = kAICPU; in RegOp()
97 OpImplyType imply_type = kCPU; in RegOp()
202 bool OpLib::DecodeOpInfo(const nlohmann::json &obj, const mindspore::kernel::OpImplyType imply_type, in DecodeOpInfo()
254 bool OpLib::DecodeAttr(const nlohmann::json &obj, const OpImplyType imply_type, in DecodeAttr()
300 bool OpLib::DecodeInputOutput(const nlohmann::json &obj, const OpImplyType imply_type, const OpIOTy… in DecodeInputOutput()
353 std::shared_ptr<OpInfo> OpLib::FindOp(const std::string &op_name, OpImplyType imply_type, bool is_d… in FindOp()
|
| D | opinfo.h | 28 enum OpImplyType { kAKG = 0, kTBE = 1, kAICPU = 2, kCPU }; enum
127 OpImplyType imply_type() const { return imply_type_; } in imply_type()
145 void set_imply_type(const OpImplyType imply_type) { imply_type_ = imply_type; } in set_imply_type()
174 OpImplyType imply_type_ = kTBE;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_metadata.cc | 31 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAKG); in AkgMetadataInfo()
|
| D | akg_kernel_json_generator.cc | 53 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/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 35 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in AicpuMetadataInfo()
|
| D | aicpu_kernel_build.cc | 155 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in SetNodeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 255 const auto &op_info = kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kCPU); in IsDynamicParamKernel()
321 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kCPU); in SetKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 81 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/cpu/ |
| D | cpu_kernel_factory.cc | 128 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(kernel_name, kernel::OpImplyType::kCPU); in CPUKernelAttrCheck()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_dynaminc_shape_util.cc | 371 return mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kTBE, is_dynamic_shape); in FindOp()
|