Home
last modified time | relevance | path

Searched full:permute_copy (Results 1 – 25 of 58) sorted by relevance

123

/aosp_15_r20/external/executorch/backends/cadence/aot/tests/
H A Dtest_reorder_ops_passes.py153 exir_ops.edge.aten.permute_copy.default,
162 exir_ops.edge.aten.permute_copy.default,
188 exir_ops.edge.aten.permute_copy.default,
195 exir_ops.edge.aten.permute_copy.default,
286 self.assertTrue(nodes[2] == exir_ops.edge.aten.permute_copy)
287 self.assertTrue(nodes[3] == exir_ops.edge.aten.permute_copy)
319 self.assertTrue(nodes[2] == exir_ops.edge.aten.permute_copy)
320 self.assertTrue(nodes[3] == exir_ops.edge.aten.permute_copy)
353 self.assertTrue(nodes[1] == exir_ops.edge.aten.permute_copy)
355 self.assertTrue(nodes[3] == exir_ops.edge.aten.permute_copy)
H A Dtest_replace_ops_passes.py1121 count_node(graph_after_passes, exir_ops.edge.aten.permute_copy.default), 0
1148 count_node(graph_after_passes, exir_ops.edge.aten.permute_copy.default), 0
1174 count_node(graph_after_passes, exir_ops.edge.aten.permute_copy.default), 0
1400 self.assertEqual(count_node(gm, exir_ops.edge.aten.permute_copy.default), 0)
1411 count_node(gm_after_replacement, exir_ops.edge.aten.permute_copy.default),
1436 count_node(gm_after_replacement, exir_ops.edge.aten.permute_copy.default),
1505 self.assertEqual(count_node(gm, exir_ops.edge.aten.permute_copy.default), 0)
1519 count_node(gm_after_replacement, exir_ops.edge.aten.permute_copy.default),
1548 self.assertEqual(count_node(gm, exir_ops.edge.aten.permute_copy.default), 0)
H A Dtest_remove_ops_passes.py548 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 0
574 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 0
609 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 0
641 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 2
673 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 2
H A Dtest_fusion_ops_passes.py193 count_node(graph_module, exir_ops.edge.aten.permute_copy.default), 1
450 op=exir_ops.edge.aten.permute_copy.default,
464 exir_ops.edge.aten.permute_copy.default: 1,
476 exir_ops.edge.aten.permute_copy.default: 0,
/aosp_15_r20/external/executorch/backends/qualcomm/_passes/
H A Dconvert_to_linear.py33 permute_copy = exir_ops.edge.aten.permute_copy.default variable in ConvertToLinear
42 {view_copy: 2, permute_copy: 1, addmm: 1},
43 {permute_copy: 1, addmm: 1},
47 {view_copy: 3, permute_copy: 1, expand_copy: 2, add: 1, bmm: 1},
48 {view_copy: 3, permute_copy: 1, expand_copy: 2, bmm: 1},
52 {view_copy: 2, permute_copy: 1, mm: 1},
53 {permute_copy: 1, mm: 1},
H A Dfuse_consecutive_transpose.py25 exir_ops.edge.aten.permute_copy.default,
60 permute_op = exir_ops.edge.aten.permute_copy.default
H A Dlayout_transform.py148 node.target == exir_ops.edge.aten.permute_copy.default
181 exir_ops.edge.aten.permute_copy.default,
H A Drecompose_pixel_unshuffle.py20 self.permute_target = exir_ops.edge.aten.permute_copy.default
/aosp_15_r20/external/executorch/backends/example/example_operators/
H A Ddropout.py22 permute_copy = torch.ops.aten.permute_copy.default(fn_weight, [1, 0]); fn_weight = None
23 …ddmm = torch.ops.aten.addmm.default(fn_bias, arg2_1, permute_copy); fn_bias = arg2_1 = permute_co…
H A Dadaptive_avg_pool2d.py22 permute_copy = torch.ops.aten.permute_copy.default(fn_weight, [1, 0]); fn_weight = None
23 …ddmm = torch.ops.aten.addmm.default(fn_bias, arg2_1, permute_copy); fn_bias = arg2_1 = permute_co…
H A Dlinear.py22 permute_copy = torch.ops.aten.permute_copy.default(fn_weight, [1, 0]); fn_weight = None
23 …ddmm = torch.ops.aten.addmm.default(fn_bias, arg2_1, permute_copy); fn_bias = arg2_1 = permute_co…
/aosp_15_r20/external/executorch/backends/xnnpack/test/ops/
H A Dpermute.py31 z = torch.permute_copy(y, self.dims)
60 .check_count({"torch.ops.aten.permute_copy.default": 1})
102 torch.ops.aten.permute_copy.default: 1,
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dreorder_ops.py36 torch.ops.aten.permute_copy,
49 exir_ops.edge.aten.permute_copy,
691 exir_ops.edge.aten.permute_copy: "default",
705 # Transform only for pattern permute_copy->view_copy, and
706 # view_copy op is the only user of permute_copy.
H A Dfuse_ops.py449 exir_ops.edge.aten.permute_copy.default,
494 exir_ops.edge.aten.permute_copy.default,
678 exir_ops.edge.aten.permute_copy.default,
986 exir_ops.edge.aten.permute_copy.default,
H A Dreplace_ops.py423 if op != exir_ops.edge.aten.permute_copy.default:
838 p_target = exir_ops.edge.aten.permute_copy.default
1150 exir_ops.edge.aten.permute_copy.default, (proxy, permute_indices), {}, meta
1160 exir_ops.edge.aten.permute_copy.default, (proxy, permute_indices), {}, meta
1621 exir_ops.edge.aten.permute_copy.default,
1647 elif op == exir_ops.edge.aten.permute_copy.default:
H A Dremove_ops.py700 node.target == exir_ops.edge.aten.permute_copy.default
706 node.target == exir_ops.edge.aten.permute_copy.default
/aosp_15_r20/external/executorch/backends/xnnpack/partition/
H A Dconfigs.py58 exir_ops.edge.aten.permute_copy.default,
91 exir_ops.edge.aten.permute_copy.default,
/aosp_15_r20/external/executorch/backends/transforms/
H A Daddmm_mm_to_linear.py123 ops.aten.permute_copy.default,
140 ops.aten.permute_copy.default,
/aosp_15_r20/external/executorch/docs/source/
H A Dcompiler-custom-compiler-passes.md524 …%permute_default : [#users=1] = call_function[target=torch.ops.aten.permute_copy.default](args = (…
528 …%permute_default_1 : [#users=1] = call_function[target=torch.ops.aten.permute_copy.default](args =…
533 …%permute_default_2 : [#users=1] = call_function[target=torch.ops.aten.permute_copy.default](args =…
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dop_transpose.py21 target = ["aten.permute_copy.default"]
/aosp_15_r20/external/executorch/exir/passes/
H A Dreplace_broken_ops_with_function_ops_pass.py22 torch.ops.aten.permute.default: torch.ops.aten.permute_copy.default,
/aosp_15_r20/external/executorch/backends/xnnpack/operators/
H A Dop_permute.py32 target = "aten.permute_copy.default"
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_permute_copy.cpp69 ET_SWITCH_ALL_TYPES(in_type, ctx, "permute_copy.out", CTYPE, [&] { in permute_copy_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml148 - permute_copy
/aosp_15_r20/external/executorch/backends/arm/operators/
H A Dop_permute.py70 target = "aten.permute_copy.default"

123