Searched refs:kernel_key (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/torchgen/executorch/ |
| D | parse.py | 67 for kernel_key in kernel_keys:
68 assert kernel_key not in kernel_mapping, (
69 "Duplicate kernel key: " + str(kernel_key) + " " + str(e)
71 kernel_mapping[kernel_key] = backend_metadata
|
| /external/vboot_reference/firmware/lib20/ |
| D | kernel.c | 45 struct vb2_public_key kernel_key; in vb2_load_kernel_keyblock() local
69 VB2_TRY(vb2_unpack_key_buffer(&kernel_key, key_data, key_size)); in vb2_load_kernel_keyblock()
95 rv = vb2_verify_keyblock(kb, block_size, &kernel_key, &wb); in vb2_load_kernel_keyblock()
|
| /external/pytorch/torchgen/selective_build/ |
| D | selector.py | 239 def et_get_selected_kernels(self, op_name: str, kernel_key: list[str]) -> list[str]:
245 return kernel_key if self.include_all_operators else []
252 for key in kernel_key:
262 if "default" not in kernel_key:
|
| /external/pytorch/torchgen/ |
| D | gen_executorch.py | 190 kernel_key: ETKernelKey | list[ETKernelKey] = unbox_kernel_entry[1][0]
197 if not isinstance(kernel_key, list):
198 kernel_key = [kernel_key]
200 op_name, [k.to_native_string() for k in kernel_key]
303 (native_function, (kernel_key, metadata))
305 for kernel_key, metadata in kernel_index.get_kernels(native_function).items()
|
| /external/vboot_reference/firmware/2lib/ |
| D | 2load_kernel.c | 165 struct vb2_public_key kernel_key; in vb2_verify_kernel_vblock() local
176 VB2_TRY(vb2_unpack_key_buffer(&kernel_key, key_data, key_size)); in vb2_verify_kernel_vblock()
178 kernel_key.allow_hwcrypto = vb2api_hwcrypto_allowed(ctx); in vb2_verify_kernel_vblock()
189 rv = vb2_verify_keyblock(keyblock, kbuf_size, &kernel_key, wb); in vb2_verify_kernel_vblock()
283 data_key.allow_hwcrypto = kernel_key.allow_hwcrypto; in vb2_verify_kernel_vblock()
|
| /external/executorch/codegen/tools/ |
| D | gen_oplist.py | 121 kernel_key = version + "/"
129 kernel_key += f"{io_metadata.dtype};{dim_order}|"
130 op_kernel_key_list[op_name].append(kernel_key[:-1])
|
| /external/executorch/runtime/kernel/ |
| D | operator_registry.cpp | 169 KernelKey kernel_key = KernelKey(buf); in get_op_function_from_registry() local
174 if (registered_kernels[idx].kernel_key_ == kernel_key) { in get_op_function_from_registry()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | bef_thunk.cc | 302 xccl_unique_id_callback = [&](const tfrt::gpu::XcclCliqueKey& kernel_key) in CreateXcclContext()
305 for (const int64_t device : kernel_key) { in CreateXcclContext()
|
| /external/pytorch/torch/_inductor/ |
| D | wrapper_benchmark.py | 80 for kernel_key, kernel_mod in PyCodeCache.cache.items():
|
| /external/toolchain-utils/afdo_tools/ |
| D | update_kernel_afdo.py | 471 for kernel_key, val in raw_contents.items():
472 m = key_re.fullmatch(kernel_key)
|