Home
last modified time | relevance | path

Searched refs:_parse_kernel_args_num_gb (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dmetrics.py338 def _parse_kernel_args_num_gb(kernel_fn_code, kernel_category): function
392 "kernel_args_num_gb": _parse_kernel_args_num_gb(
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_metrics.py114 metrics._parse_kernel_args_num_gb(kernel_code, "pointwise"), 2.0