Home
last modified time | relevance | path

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

12

/external/executorch/backends/qualcomm/_passes/
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},
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
/external/executorch/backends/arm/_passes/
Dconvert_expand_copy_to_repeat.py21 expand_copy = exir_ops.edge.aten.expand_copy.default variable in ConvertExpandCopyToRepeatPass
25 if op != self.expand_copy:
/external/pytorch/aten/src/ATen/native/
Dts_native_functions.yaml147 - expand_copy
208 - expand_copy
DCopy.cpp35 #include <ATen/ops/expand_copy.h>
/external/executorch/exir/passes/
Dreplace_broken_ops_with_function_ops_pass.py21 torch.ops.aten.expand.default: torch.ops.aten.expand_copy.default,
/external/executorch/backends/qualcomm/builders/
Dop_expand.py20 target = ["aten.expand_copy.default"]
/external/executorch/backends/arm/quantizer/quantization_annotation/
Dgeneric_annotator.py33 torch.ops.aten.expand_copy.default,
/external/executorch/backends/arm/operator_support/
Dtosa_supported_operators.py77 exir_ops.edge.aten.expand_copy.default,
/external/executorch/backends/cadence/aot/
Dfunctions.yaml65 - op: expand_copy.out
Dremove_ops.py132 exir_ops.edge.aten.expand_copy,
/external/executorch/kernels/aten/
Dfunctions.yaml154 - op: expand_copy.out
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesViews.cpp578 …VMAP_SUPPORT(expand_copy, SINGLE_ARG(expand_batch_rule<decltype(&ATEN_FN(expand_copy)), &ATEN_FN(e… in TORCH_LIBRARY_IMPL()
/external/executorch/backends/apple/mps/operators/
Dshape_ops.py57 target = ["aten.view_copy.default", "aten.expand_copy.default"]
/external/executorch/backends/apple/mps/runtime/operations/
DShapeOps.mm77 name:@"expand_copy"];
/external/pytorch/test/edge/
Dselected_operators.yaml158 aten::expand_copy.out:
/external/executorch/backends/cadence/aot/tests/
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
/external/pytorch/aten/src/ATen/core/
Dinterned_strings.h55 _(prim, expand_copy) \
/external/pytorch/test/
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
/external/pytorch/torch/_decomp/
D__init__.py305 aten.expand_copy,
/external/executorch/kernels/portable/
Dfunctions.yaml342 - op: expand_copy.out
/external/pytorch/test/distributed/_tensor/
Dtest_dtensor_ops.py146 xfail("expand_copy"),
/external/executorch/exir/dialects/edge/op/
Dsample_input.py468 "expand_copy.default": { # (Tensor self, SymInt[] size, *, bool implicit=False) -> Tensor
/external/pytorch/tools/autograd/
Dgen_variable_type.py192 "expand_copy",
/external/executorch/exir/dialects/edge/
Dedge.yaml3276 - func: aten::expand_copy
3278 inherits: aten::expand_copy

12