Searched refs:arg_key (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | mkl_cpu_kernel.cc | 141 void MKLCPUKernel::AddArgument(int arg_key, const dnnl::memory::desc &mem_desc, bool alloc) { in AddArgument() argument
142 arguments_[arg_key] = MKLKernelEngine::Get().CreateMemory(mem_desc, alloc); in AddArgument()
145 void MKLCPUKernel::SetArgumentHandle(int arg_key, void *ptr) { in SetArgumentHandle() argument
146 auto arg_iter = arguments_.find(arg_key); in SetArgumentHandle()
|
| D | mkl_cpu_kernel.h | 41 void AddArgument(int arg_key, const dnnl::memory::desc &mem_desc, bool alloc = false);
42 void SetArgumentHandle(int arg_key, void *ptr);
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 487 def _check_target_specific_cfgs(device, arg_key): argument
503 if not arg_key in device_cfgs:
505 supported_devices = device_cfgs[arg_key]
|
| D | _checkparam.py | 516 arg_key, arg_val = arg
518 Validator.check_subclass(arg_key, elem_type, valid_values, prim_name)
519 return (arg_key, elem_type)
555 arg_key, arg_val = arg
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | ascend_kernel_compile.cc | 116 const std::string arg_key = it.key(); in ReportToErrorManager() local
120 const auto &arg_value = GetJsonValue<std::string>(exception_message, arg_key); in ReportToErrorManager()
121 arg_map[arg_key] = arg_value; in ReportToErrorManager()
|