Home
last modified time | relevance | path

Searched full:structured_delegate (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml425 structured_delegate: sgn.out
434 structured_delegate: sgn.out
511 structured_delegate: acos.out
517 structured_delegate: acos.out
549 structured_delegate: add.out
562 structured_delegate: add.out
629 structured_delegate: addmv.out
633 structured_delegate: addmv.out
695 structured_delegate: all.out
700 structured_delegate: all.dims_out
[all …]
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py538 # delegate to the out function using the structured_delegate keyword.
545 structured_delegate: OperatorName | None
659 structured_delegate_s = e.pop("structured_delegate", None)
667 structured_delegate: OperatorName | None = None
669 structured_delegate = OperatorName.parse(structured_delegate_s)
779 structured_delegate
788 elif not structured and structured_delegate is None:
878 if structured_delegate:
918 if structured_delegate is not None:
921 f"if structured_delegate, then must not have {key} in dispatch dictionary "
[all …]
H A Dgen.py261 if f.structured_delegate is not None:
262 delegate_func = func_map.get(f.structured_delegate)
264 f"{f.func.name} is marked as a structured_delegate pointing to "
265 f"{f.structured_delegate}, but {f.structured_delegate} is missing."
268 f"{f.func.name} is marked as a structured_delegate pointing to "
269 … f"{f.structured_delegate}, but {f.structured_delegate} is not marked as structured. "
338 if f.structured_delegate is not None or backend_index.has_kernel(f):
339 # TODO: for ops with structured_delegate it should check the dispatch table of
486 f.structured_delegate is not None
1290 return (f.structured or f.structured_delegate is not None) and (
[all …]
H A Dgen_aoti_c_shim.py322 func.structured_delegate is not None
323 and func.structured_delegate in func_group_mapping
325 func_group_mapping[func.structured_delegate]
H A Dnative_function_generation.py344 structured_delegate=None,
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_executorch_gen.py46 structured_delegate: add.out
70 structured_delegate: mul.out
113 structured_delegate: add.out
137 structured_delegate: mul.out
H A Dtest_codegen_model.py72 structured_delegate: binop.out
79 structured_delegate: unop.out
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dlazy_ir.py510 is_structured = func.structured or func.structured_delegate is not None
681 is_structured = f.structured or f.structured_delegate is not None