| /aosp_15_r20/external/pytorch/torchgen/selective_build/ |
| H A D | selector.py | 47 kernel_metadata: dict[str, list[str]]
81 "kernel_metadata",
110 kernel_metadata = {}
111 kernel_metadata_dict = data.get("kernel_metadata", {})
115 kernel_metadata[str(k)] = [str(dtype) for dtype in v]
135 kernel_metadata,
235 kernel_tag in self.kernel_metadata
236 and dtype in self.kernel_metadata[kernel_tag]
282 ret["kernel_metadata"] = {
283 k: sorted(v) for (k, v) in self.kernel_metadata.items()
[all …]
|
| /aosp_15_r20/external/pytorch/tools/code_analyzer/ |
| H A D | gen_operators_yaml.py | 51 # kernel_metadata:
303 if "kernel_metadata" in model_info:
304 all_kernel_metadata.append(model_info["kernel_metadata"])
491 kernel_metadata = {}
493 kernel_metadata = merge_kernel_metadata(kernel_metadata, kt)
494 output["kernel_metadata"] = kernel_metadata
619 output["kernel_metadata"] = {}
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | update_production_ops.py | 13 kernel_metadata = {} variable
27 for kernal, dtypes in info["kernel_metadata"].items():
29 kernel_metadata[kernal] if kernal in kernel_metadata else []
31 kernel_metadata[kernal] = list(set(new_dtypes))
|
| /aosp_15_r20/external/executorch/codegen/tools/ |
| H A D | gen_selected_op_variants.py | 76 for kernel_metadata in kernel_metadata_str:
77 if kernel_metadata == "default" or "/" not in kernel_metadata:
80 x = kernel_metadata.split("/")[1]
|
| H A D | gen_oplist.py | 181 output["kernel_metadata"] = {}
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 280 KernelMetadata kernel_metadata; in GetKernel() local
281 TF_RETURN_IF_ERROR(GetKernelMetadata(rocm_kernel, &kernel_metadata)); in GetKernel()
282 kernel->set_metadata(kernel_metadata); in GetKernel()
288 KernelMetadata* kernel_metadata) { in GetKernelMetadata() argument
291 kernel_metadata->set_registers_per_thread(value); in GetKernelMetadata()
294 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_selective_build.py | 199 kernel_metadata:
226 kernel_metadata:
238 kernel_metadata:
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 288 KernelMetadata kernel_metadata; in GetKernel() local
289 TF_RETURN_IF_ERROR(GetKernelMetadata(cuda_kernel, &kernel_metadata)); in GetKernel()
290 kernel->set_metadata(kernel_metadata); in GetKernel()
428 KernelMetadata* kernel_metadata) { in GetKernelMetadata() argument
432 kernel_metadata->set_registers_per_thread(value); in GetKernelMetadata()
437 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()
|
| /aosp_15_r20/external/executorch/codegen/tools/test/ |
| H A D | test_gen_selected_op_variants.py | 39 kernel_metadata: {}
102 kernel_metadata: {}
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | metrics.py | 242 "kernel_metadata",
376 get_metric_table("kernel_metadata").add_row(
|
| /aosp_15_r20/external/pytorch/tools/lite_interpreter/ |
| H A D | gen_selected_mobile_ops_header.py | 73 for kernel_tag, dtypes in selective_builder.kernel_metadata.items():
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
| H A D | tracer.cpp | 190 yaml_out << "kernel_metadata:"; in main()
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 5 kernel_metadata: {}
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| H A D | gpu_executor.h | 338 KernelMetadata* kernel_metadata);
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | halide.py | 1696 if is_metric_table_enabled("kernel_metadata"):
|
| H A D | triton.py | 3057 if is_metric_table_enabled("kernel_metadata"):
|
| /aosp_15_r20/external/bcc/libbpf-tools/powerpc/ |
| H A D | vmlinux.h | 38835 u64 kernel_metadata; member
|
| H A D | vmlinux_600.h | 38835 u64 kernel_metadata; member
|