| /aosp_15_r20/external/executorch/docs/source/ |
| H A D | debug-backend-delegate.md | 54 …rs=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.unsqueeze_copy.default](args …
55 …rs=1] = call_function[target=executorch.exir.dialects.edge._ops.aten.unsqueeze_copy.default](args …
|
| /aosp_15_r20/external/executorch/backends/transforms/ |
| H A D | view_copy_to_squeeze_unsqueeze.py | 20 … Replaces view_copy nodes with unsqueeze_copy.default nodes if the view node adds a dim of size 1.
28 exir_ops.edge.aten.unsqueeze_copy.default
|
| /aosp_15_r20/external/executorch/kernels/aten/ |
| H A D | functions.yaml | 402 - op: unsqueeze_copy.dim_out
404 - op: unsqueeze_copy.out
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | shape_ops.py | 115 target = ["aten.unsqueeze_copy.default", "aten.squeeze_copy.dims"]
127 if node.target is exir_ops.edge.aten.unsqueeze_copy.default
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | FunctionalInverses.cpp | 13 // will all call `permute/unsqueeze_copy()` instead of `permute/unsqueeze`.
39 result = at::unsqueeze_copy(result, dim); in unsqueeze_copy_to()
66 result = at::unsqueeze_copy(result, d); in unsqueeze_copy_to()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_functionalization.py | 1278 … unsqueeze_copy = torch.ops.aten.unsqueeze_copy.default(transpose_copy, 0); transpose_copy = None
1279 squeeze_copy = torch.ops.aten.squeeze_copy.default(unsqueeze_copy); unsqueeze_copy = None
1287 …unsqueeze_copy_1 = torch.ops.aten.unsqueeze_copy.default(transpose_copy_1, 0); transpose_copy_1 =…
1290 … unsqueeze_copy_2 = torch.ops.aten.unsqueeze_copy.default(slice_scatter, 0); slice_scatter = None
1298 …unsqueeze_copy_3 = torch.ops.aten.unsqueeze_copy.default(transpose_copy_3, 0); transpose_copy_3 =…
1311 …unsqueeze_copy_4 = torch.ops.aten.unsqueeze_copy.default(transpose_copy_4, 0); transpose_copy_4 =…
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/tests/ |
| H A D | test_remove_ops_passes.py | 599 z = torch.unsqueeze_copy(z, 0)
631 z = torch.unsqueeze_copy(z, 0)
|
| H A D | test_replace_ops_passes.py | 813 # First, replace the aten.unsqueeze_copy with an aten.view_copy op
819 # Assert that all the aten.unsqueeze_copy nodes are removed.
825 count_node(graph_after_passes, exir_ops.aten.unsqueeze_copy),
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_unsqueeze.py | 18 target = ["aten.unsqueeze_copy.default"]
|
| /aosp_15_r20/external/executorch/backends/arm/operators/ |
| H A D | op_unsqueeze.py | 23 target = "aten.unsqueeze_copy.default"
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Unsqueeze.cpp | 49 VK_REGISTER_OP(aten.unsqueeze_copy.default, unsqueeze);
|
| /aosp_15_r20/external/executorch/exir/passes/ |
| H A D | replace_broken_ops_with_function_ops_pass.py | 24 torch.ops.aten.unsqueeze.default: torch.ops.aten.unsqueeze_copy.default,
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | reorder_ops.py | 40 torch.ops.aten.unsqueeze_copy,
51 exir_ops.edge.aten.unsqueeze_copy,
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_unsqueeze_copy.cpp | 21 // unsqueeze_copy.out(Tensor self, int dim, *, Tensor(a!) out) -> Tensor(a!)
|
| /aosp_15_r20/external/executorch/backends/arm/test/quantizer/ |
| H A D | test_generic_annotater.py | 54 SingleOpModel(torch.unsqueeze_copy, (torch.rand(8, 8),), dim=0)
|
| /aosp_15_r20/external/executorch/backends/xnnpack/operators/ |
| H A D | op_squeeze.py | 86 target = "aten.unsqueeze_copy.default"
|
| /aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/ |
| H A D | generic_annotator.py | 30 torch.ops.aten.unsqueeze_copy.default,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ts_native_functions.yaml | 163 - unsqueeze_copy
|
| /aosp_15_r20/external/executorch/backends/arm/operator_support/ |
| H A D | tosa_supported_operators.py | 110 exir_ops.edge.aten.unsqueeze_copy.default,
|
| /aosp_15_r20/external/executorch/backends/vulkan/partitioner/ |
| H A D | supported_ops.py | 107 exir_ops.edge.aten.unsqueeze_copy.default,
|
| /aosp_15_r20/external/executorch/backends/arm/_passes/ |
| H A D | conv1d_unsqueeze_pass.py | 126 graph, exir_ops.edge.aten.unsqueeze_copy.default
|
| H A D | annotate_channels_last_dim_order_pass.py | 153 elif node.target == exir_ops.edge.aten.unsqueeze_copy.default:
|
| /aosp_15_r20/external/executorch/backends/xnnpack/_passes/ |
| H A D | conv1d_unsqueeze_pass.py | 155 graph, exir_ops.edge.aten.unsqueeze_copy.default
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 422 aten::unsqueeze_copy.out:
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | interned_strings.h | 53 _(prim, unsqueeze_copy) \
|