Searched defs:arg_key (Results 1 – 2 of 2) sorted by relevance
200 void MKLCpuKernelMod::AddArgument(int arg_key, const dnnl::memory::desc &mem_desc, bool alloc) { in AddArgument()208 void MKLCpuKernelMod::SetArgumentHandle(int arg_key, void *ptr) { in SetArgumentHandle()
1169 def _check_target_specific_cfgs(device, arg_key): argument