Home
last modified time | relevance | path

Searched full:expand_copy (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/qualcomm/_passes/
H A Dconvert_to_linear.py34 expand_copy = exir_ops.edge.aten.expand_copy.default variable in ConvertToLinear
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},
178 # weight -> expand_copy -> view_copy -> input of bmm
180 # input -> expand_copy -> view_copy -> input of bmm
H A Dconvert_bmm_to_matmul.py23 expand_copy = exir_ops.edge.aten.expand_copy.default variable in ConvertBmmToMatmul
28 {expand_copy: 2, view_copy: 3, bmm: 1},
29 {expand_copy: 2, view_copy: 3, bmm: 1, clone: 1},
/aosp_15_r20/external/executorch/backends/arm/_passes/
H A Dconvert_expand_copy_to_repeat.py21 expand_copy = exir_ops.edge.aten.expand_copy.default variable in ConvertExpandCopyToRepeatPass
25 if op != self.expand_copy:
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml147 - expand_copy
208 - expand_copy
H A DCopy.cpp35 #include <ATen/ops/expand_copy.h>
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp578 …VMAP_SUPPORT(expand_copy, SINGLE_ARG(expand_batch_rule<decltype(&ATEN_FN(expand_copy)), &ATEN_FN(e… in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/
H A Dtest_functionalization.py255expand_copy = torch.ops.aten.expand_copy.default(ones_like, [16, 64, 128, 128]); ones_like = None
256 …view_copy_5 = torch.ops.aten.view_copy.default(expand_copy, [1, 1024, 128, 128]); expand_copy = N…
1616 expand_copy = torch.ops.aten.expand_copy.default(arg0_1, [2, 2]); arg0_1 = None
1617 return expand_copy
/aosp_15_r20/external/executorch/backends/cadence/aot/tests/
H A Dtest_remove_ops_passes.py234 return torch.ops.aten.expand_copy(x, [2, 3, 5])
242 count_node(graph_module, exir_ops.edge.aten.expand_copy.default), 0
/aosp_15_r20/external/executorch/exir/passes/
H A Dreplace_broken_ops_with_function_ops_pass.py21 torch.ops.aten.expand.default: torch.ops.aten.expand_copy.default,
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dop_expand.py20 target = ["aten.expand_copy.default"]
/aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/
H A Dgeneric_annotator.py33 torch.ops.aten.expand_copy.default,
/aosp_15_r20/external/executorch/backends/arm/operator_support/
H A Dtosa_supported_operators.py77 exir_ops.edge.aten.expand_copy.default,
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml154 - op: expand_copy.out
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dfunctions.yaml65 - op: expand_copy.out
H A Dremove_ops.py132 exir_ops.edge.aten.expand_copy,
/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dshape_ops.py57 target = ["aten.view_copy.default", "aten.expand_copy.default"]
/aosp_15_r20/external/executorch/backends/apple/mps/runtime/operations/
H A DShapeOps.mm77 name:@"expand_copy"];
/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml158 aten::expand_copy.out:
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dinterned_strings.h55 _(prim, expand_copy) \
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py305 aten.expand_copy,
/aosp_15_r20/external/executorch/kernels/portable/
H A Dfunctions.yaml342 - op: expand_copy.out
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_dtensor_ops.py146 xfail("expand_copy"),
/aosp_15_r20/external/executorch/exir/dialects/edge/op/
H A Dsample_input.py468 "expand_copy.default": { # (Tensor self, SymInt[] size, *, bool implicit=False) -> Tensor
/aosp_15_r20/external/executorch/exir/dialects/edge/
H A Dedge.yaml3276 - func: aten::expand_copy
3278 inherits: aten::expand_copy
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_variable_type.py192 "expand_copy",

12