Searched refs:_parse_numel (Results 1 – 1 of 1) sorted by relevance
330 def _parse_numel(proper_kernel_fn_code, numel_arg_name): function389 "xnumel": _parse_numel(proper_kernel_fn_code, "xnumel"),390 "ynumel": _parse_numel(proper_kernel_fn_code, "ynumel"),391 "rnumel": _parse_numel(proper_kernel_fn_code, "rnumel"),