Home
last modified time | relevance | path

Searched full:transpose_copy (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/executorch/backends/cadence/aot/tests/
H A Dtest_fusion_ops_passes.py196 count_node(graph_module, exir_ops.edge.aten.transpose_copy.int), 0
446 op=exir_ops.edge.aten.transpose_copy.int,
463 exir_ops.edge.aten.transpose_copy.int: 1,
475 exir_ops.edge.aten.transpose_copy.int: 0,
488 op=exir_ops.edge.aten.transpose_copy.int,
496 op=exir_ops.edge.aten.transpose_copy.int,
504 exir_ops.edge.aten.transpose_copy.int: 2,
514 exir_ops.edge.aten.transpose_copy.int: 0,
524 op=exir_ops.edge.aten.transpose_copy.int,
532 op=exir_ops.edge.aten.transpose_copy.int,
[all …]
H A Dtest_replace_ops_passes.py333 count_node(graph_after_passes, exir_ops.edge.aten.transpose_copy.int),
1094 count_node(graph_after_passes, exir_ops.edge.aten.transpose_copy.int), 0
1151 count_node(graph_after_passes, exir_ops.edge.aten.transpose_copy.int), 1
1177 count_node(graph_after_passes, exir_ops.edge.aten.transpose_copy.int), 0
1323 self.assertEqual(count_node(gm, exir_ops.edge.aten.transpose_copy.int), 0)
1334 count_node(gm_after_replacement, exir_ops.edge.aten.transpose_copy.int),
1359 count_node(gm_after_replacement, exir_ops.edge.aten.transpose_copy.int),
1584 count_node(gm_after_pass, exir_ops.edge.aten.transpose_copy.int),
1601 count_node(gm_after_pass, exir_ops.edge.aten.transpose_copy.int),
1617 count_node(gm_after_pass, exir_ops.edge.aten.transpose_copy.int),
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_functionalization.py946 transpose_copy = torch.ops.aten.transpose_copy.int(arg0_1, 1, 0)
947 slice_copy = torch.ops.aten.slice_copy.Tensor(transpose_copy, 0, 0, 2); transpose_copy = None
949 transpose_copy_1 = torch.ops.aten.transpose_copy.int(arg0_1, 1, 0); arg0_1 = None
951 transpose_copy_2 = torch.ops.aten.transpose_copy.int(slice_scatter, 1, 0); slice_scatter = None
952 transpose_copy_3 = torch.ops.aten.transpose_copy.int(transpose_copy_2, 1, 0)
954 …transpose_copy_4 = torch.ops.aten.transpose_copy.int(transpose_copy_2, 1, 0); transpose_copy_2 = …
1003 transpose_copy = torch.ops.aten.transpose_copy.int(arg0_1, 1, 0)
1004 select_copy = torch.ops.aten.select_copy.int(transpose_copy, 0, 0); transpose_copy = None
1006 transpose_copy_1 = torch.ops.aten.transpose_copy.int(arg0_1, 1, 0); arg0_1 = None
1008 … transpose_copy_2 = torch.ops.aten.transpose_copy.int(select_scatter, 1, 0); select_scatter = None
[all …]
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dreplace_ops.py273 Replace t_copy with transpose_copy.int. If the input is 1D, the t_copy is
291 exir_ops.edge.aten.transpose_copy.int, transpose_args, kwargs, meta
364 and mat2.target == exir_ops.edge.aten.transpose_copy.int
439 exir_ops.edge.aten.transpose_copy.int, new_args, kwargs, meta
707 # graph operation (in this case a transpose_copy op) to be an explicit
712 exir_ops.edge.aten.transpose_copy.int,
839 t_target = exir_ops.edge.aten.transpose_copy.int
1137 exir_ops.edge.aten.transpose_copy.int, (proxy, dim0, dim1), {}, meta
1426 exir_ops.edge.aten.transpose_copy.int,
1595 exir_ops.edge.aten.transpose_copy.int,
[all …]
H A Dfuse_ops.py448 exir_ops.edge.aten.transpose_copy.int,
479 if tp.target == exir_ops.edge.aten.transpose_copy.int
677 exir_ops.edge.aten.transpose_copy.int,
970 producer_op_packets={exir_ops.edge.aten.transpose_copy},
971 consumer_op_packets={exir_ops.edge.aten.transpose_copy},
985 exir_ops.edge.aten.transpose_copy.int,
H A Dreorder_ops.py34 torch.ops.aten.transpose_copy,
48 exir_ops.edge.aten.transpose_copy,
H A Dcompiler_utils.py115 assert node.target == exir_ops.edge.aten.transpose_copy.int
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_transpose_copy.cpp25 * transpose_copy.int_out(Tensor self, int dim0, int dim1, *, Tensor(a!) out)
/aosp_15_r20/external/executorch/backends/arm/test/quantizer/
H A Dtest_generic_annotater.py77 SingleOpModel(torch.transpose_copy, (torch.randn(2, 3),), dim0=0, dim1=1),
/aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/
H A Dgeneric_annotator.py52 torch.ops.aten.transpose_copy.int,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml162 - transpose_copy.int
H A Dnative_functions.yaml14519 - func: transpose_copy.int(Tensor self, int dim0, int dim1) -> Tensor
14524 autogen: transpose_copy.int_out
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml392 - op: transpose_copy.int_out
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect1304 aten::transpose_copy.int
1305 aten::transpose_copy.int_out
/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml404 aten::transpose_copy.int_out:
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dinterned_strings.h52 _(prim, transpose_copy) \
/aosp_15_r20/external/executorch/kernels/portable/
H A Dfunctions.yaml890 - op: transpose_copy.int_out
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalInverses.cpp297 return transpose_copy(mutated_view, dim0, dim1); in transpose_int_inverse()
/aosp_15_r20/external/executorch/exir/dialects/edge/op/
H A Dsample_input.py1194 "transpose_copy.int": { # (Tensor self, int dim0, int dim1) -> Tensor
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py4809 transpose_copy = torch.ops.aten.transpose_copy.int(x_1, 1, 0); x_1 = None
4810 return transpose_copy
/aosp_15_r20/external/executorch/exir/dialects/edge/
H A Dedge.yaml7053 - func: aten::transpose_copy.int
7055 inherits: aten::transpose_copy.int
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py2164 "torch.transpose_copy",
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py1280 torch.transpose_copy: lambda self, dim0, dim1: -1,
H A D_torch_docs.py13895 torch.transpose_copy,