Searched defs:arg_key (Results 1 – 3 of 3) sorted by relevance
141 void MKLCPUKernel::AddArgument(int arg_key, const dnnl::memory::desc &mem_desc, bool alloc) { in AddArgument()145 void MKLCPUKernel::SetArgumentHandle(int arg_key, void *ptr) { in SetArgumentHandle()
487 def _check_target_specific_cfgs(device, arg_key): argument
116 const std::string arg_key = it.key(); in ReportToErrorManager() local