Home
last modified time | relevance | path

Searched full:unbind_copy (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_unbind_copy.cpp23 * unbind_copy.int_out(Tensor input, int dim=0, *, Tensor(a!)[] out) -> ()
58 Bool, in_type, ctx, "unbind_copy.int_out", CTYPE_IN, [&]() { in unbind_copy_int_out()
60 Bool, out_type, ctx, "unbind_copy.int_out", CTYPE_OUT, [&]() { in unbind_copy_int_out()
/aosp_15_r20/external/pytorch/test/
H A Dtest_functionalization.py1061 unbind_copy = torch.ops.aten.unbind_copy.int(transpose_copy); transpose_copy = None
1062 getitem = unbind_copy[0]
1063 getitem_1 = unbind_copy[1]; unbind_copy = getitem_1 = None
1069 unbind_copy_1 = torch.ops.aten.unbind_copy.int(transpose_copy_3); transpose_copy_3 = None
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml157 #- unbind_copy.int
H A Dnative_functions.yaml14589 - func: unbind_copy.int(Tensor self, int dim=0) -> Tensor[]
14595 - func: unbind_copy.int_out(Tensor self, int dim=0, *, Tensor(a!)[] out) -> ()
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml398 - op: unbind_copy.int_out
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect1308 aten::unbind_copy.int
1309 aten::unbind_copy.int_out
/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml410 aten::unbind_copy.int_out:
/aosp_15_r20/external/executorch/kernels/portable/
H A Dfunctions.yaml905 - op: unbind_copy.int_out
/aosp_15_r20/external/executorch/exir/dialects/edge/op/
H A Dsample_input.py1213 "unbind_copy.int": { # (Tensor self, int dim=0) -> Tensor[]
/aosp_15_r20/external/executorch/exir/dialects/edge/
H A Dedge.yaml7071 - func: aten::unbind_copy.int
7073 inherits: aten::unbind_copy.int
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py2177 "torch.unbind_copy",
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py1290 torch.unbind_copy: lambda self, dim=0: -1,
H A D_torch_docs.py13943 torch.unbind_copy,