Home
last modified time | relevance | path

Searched full:select_copy (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_select.py39 return torch.select_copy(x, dim=dim, index=index)
155 self.SelectCopy(), test_data, export_target="torch.ops.aten.select_copy.int"
167 self.SelectCopy(), test_data, export_target="torch.ops.aten.select_copy.int"
179 self.SelectCopy(), test_data, export_target="torch.ops.aten.select_copy.int"
191 self.SelectCopy(), test_data, export_target="torch.ops.aten.select_copy.int"
/aosp_15_r20/external/executorch/backends/cadence/aot/tests/
H A Dtest_remove_ops_passes.py386 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 1
391 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 0
406 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 1
411 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 0
426 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 1
431 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 0
446 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 1
451 count_node(graph_module, exir_ops.edge.aten.select_copy.int), 0
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/unordered/test/unordered/
Dcopy_tests.cpp163 test::cxx11_allocator<test::object, test::select_copy> >*
166 test::cxx11_allocator<test::object, test::select_copy> >*
169 test::cxx11_allocator<test::object, test::select_copy> >*
172 test::equal_to, test::cxx11_allocator<test::object, test::select_copy> >*
/aosp_15_r20/external/pytorch/test/
H A Dtest_functionalization.py1004 select_copy = torch.ops.aten.select_copy.int(transpose_copy, 0, 0); transpose_copy = None
1005 add = torch.ops.aten.add.Tensor(select_copy, ones); select_copy = ones = None
1010 …select_copy_1 = torch.ops.aten.select_copy.int(transpose_copy_3, 0, 0); transpose_copy_3 = select…
1303 … select_copy = torch.ops.aten.select_copy.int(view_copy_1, 0, 0); view_copy_1 = select_copy = None
1307 select_copy_1 = torch.ops.aten.select_copy.int(view_copy_10, 0, 0); view_copy_10 = None
1884 select_copy = torch.ops.aten.select_copy.int(zeros, 0, 5)
1885 fill = torch.ops.aten.fill.Scalar(select_copy, 1); select_copy = None
1887 select_copy_1 = torch.ops.aten.select_copy.int(select_scatter, 0, 5); select_copy_1 = None
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dreorder_ops.py44 torch.ops.aten.select_copy,
55 exir_ops.edge.aten.select_copy,
61 torch.ops.aten.select_copy,
63 exir_ops.edge.aten.select_copy,
558 exir_ops.edge.aten.select_copy,
H A Dremove_ops.py303 exir_ops.edge.aten.select_copy.int,
323 if sel_node.target != exir_ops.edge.aten.select_copy.int:
/aosp_15_r20/external/executorch/backends/arm/operators/
H A Dop_select.py25 target = "aten.select_copy.int"
47 # For aten.select_copy, the output will be rank[input_shape - 1]
/aosp_15_r20/external/executorch/backends/vulkan/_passes/
H A Dremove_local_scalar_dense_ops.py23 and node.target == exir_ops.edge.aten.select_copy.int
71 and replace_node.target == exir_ops.edge.aten.select_copy.int
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp49 #include <ATen/ops/select_copy.h>
969 constexpr const char* select_name = (require_view ? "select()" : "select_copy()"); in select_sparse_csr_worker()
993 return at::select_copy(self, dim, index); in select_sparse_csr_worker()
1114 computing the result of torch.select_copy(input, dim, in select_sparse_csr_worker()
1223 ": no view exists for the given input, consider using torch.select_copy."); in select_sparse_csr_worker()
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dop_select_copy.py21 target = ["aten.select_copy.int", "aten.select.int"]
/aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/
H A Dgeneric_annotator.py45 torch.ops.aten.select_copy.int,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml150 - select_copy.int
/aosp_15_r20/external/executorch/backends/arm/operator_support/
H A Dtosa_supported_operators.py101 exir_ops.edge.aten.select_copy.int,
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DSelect.cpp132 VK_REGISTER_OP(aten.select_copy.int, select_int);
/aosp_15_r20/external/executorch/backends/vulkan/partitioner/
H A Dsupported_ops.py116 exir_ops.edge.aten.select_copy.int,
H A Dvulkan_partitioner.py195 if node.target == exir_ops.edge.aten.select_copy.int:
/aosp_15_r20/external/executorch/exir/emit/
H A D_emitter.py722 name="aten::select_copy",
773 # update the select_copy kernel call to output to those id's.
818 # not jump back to the select_copy instruction that we generated at the beginning of this
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml344 - op: select_copy.int_out
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect1145 aten::select_copy.int
1146 aten::select_copy.int_out
/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dshape_ops.py149 target = "aten.select_copy.int"
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/unordered/test/objects/
Dcxx11_allocator.hpp41 struct select_copy : allocator_false struct
/aosp_15_r20/external/pytorch/test/edge/
H A Dselected_operators.yaml332 aten::select_copy.int_out:
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dop_registry.py491 exir_ops.edge.aten.select_copy.int,
/aosp_15_r20/external/executorch/exir/emit/test/
H A Dtest_emit.py655 # and select_copy, which is what we verify here. The first operator is to generate
674 "aten::select_copy",
/aosp_15_r20/external/executorch/kernels/portable/
H A Dfunctions.yaml785 - op: select_copy.int_out

12