Home
last modified time | relevance | path

Searched full:max_pool2d_with_indices (Results 1 – 25 of 108) sorted by relevance

12345

/aosp_15_r20/external/executorch/backends/xnnpack/_passes/
H A Dremove_getitem_op.py15 …This remove item is used to remove getitem operator for max_pool2d_with_indices.default operator, …
28 node.target.__name__ == "aten.max_pool2d_with_indices.default"
57 == "aten.max_pool2d_with_indices.default"
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp20 #include <ATen/ops/max_pool2d_with_indices.h>
96 auto [output, indices] = at::max_pool2d_with_indices( in max_pool1d_with_indices()
162 auto output_and_indices = at::max_pool2d_with_indices( in max_pool2d()
H A Dts_native_functions.yaml77 - max_pool2d_with_indices
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesPooling.cpp53 …elf, self_bdim, kernel_size, stride, padding, dilation, ceil_mode, 2, at::max_pool2d_with_indices); in max_pool2d_with_indices_batch_rule()
69 VMAP_SUPPORT(max_pool2d_with_indices, max_pool2d_with_indices_batch_rule); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h262 inline std::tuple<Tensor, Tensor> max_pool2d_with_indices( in max_pool2d_with_indices() function
269 return torch::max_pool2d_with_indices( in max_pool2d_with_indices()
281 /// F::max_pool2d_with_indices(x, F::MaxPool2dFuncOptions(3).stride(2));
283 inline std::tuple<Tensor, Tensor> max_pool2d_with_indices( in max_pool2d_with_indices() function
286 return detail::max_pool2d_with_indices( in max_pool2d_with_indices()
/aosp_15_r20/external/executorch/backends/xnnpack/test/ops/
H A Dmaxpool2d.py51 Note that the export process generates aten.max_pool2d_with_indices. The remove_getitem_op
86 .check_count({"torch.ops.aten.max_pool2d_with_indices.default": 1})
/aosp_15_r20/external/executorch/backends/xnnpack/partition/
H A Dconfigs.py46 exir_ops.edge.aten.max_pool2d_with_indices.default,
95 exir_ops.edge.aten.max_pool2d_with_indices.default,
/aosp_15_r20/external/pytorch/docs/source/
H A Dexport.rst198 max_pool2d_with_indices = torch.ops.aten.max_pool2d_with_indices.default(
201 getitem: f32[1, 16, 85, 85] = max_pool2d_with_indices[0];
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py4063 return aten.max_pool2d_with_indices(x, [3, 3], [2, 2])
4072 return aten.max_pool2d_with_indices(x, [3, 3], [2, 2])
4083 aten.max_pool2d_with_indices(x, [3, 3], [2, 2], [1, 1]),
4084 aten.max_pool2d_with_indices(
4106 return aten.max_pool2d_with_indices(x, [3, 3], [2, 2], [0, 0], [1, 1], True)
4115 return aten.max_pool2d_with_indices(x, [3, 3], [])
4125 return aten.max_pool2d_with_indices(x, [13, 13], [])
4151 return aten.max_pool2d_with_indices(x, [3, 2], [2, 1], [1, 1], [1, 2])
4258 max_pool2d_with_indices = torch.ops.aten.max_pool2d_with_indices(
4261 getitem = max_pool2d_with_indices[0]
[all …]
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset7.py32 "max_pool2d_with_indices",
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/representation/
H A Drewrite.py553 out_fp32, _ = torch.ops.aten.max_pool2d_with_indices.default(
581 out_i32, _ = torch.ops.aten.max_pool2d_with_indices.default(
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_max_pool2d_with_indices.cpp74 in_type, ctx, "max_pool2d_with_indices.out", CTYPE, [&]() { in max_pool2d_with_indices_out()
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py346 def max_pool2d_with_indices( function
1316 …"aten::max_pool2d_with_indices(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=0…
1317 max_pool2d_with_indices,
/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dpooling_ops.py24 target = "aten.max_pool2d_with_indices.default"
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dop_max_pool2d.py21 target = ["aten.max_pool2d_with_indices.default"]
/aosp_15_r20/external/executorch/backends/arm/operator_support/
H A Dtosa_supported_operators.py93 exir_ops.edge.aten.max_pool2d_with_indices.default,
/aosp_15_r20/external/pytorch/torchgen/aoti/
H A Dfallback_ops.py79 "aten.max_pool2d_with_indices.default",
/aosp_15_r20/external/executorch/backends/vulkan/partitioner/
H A Dsupported_ops.py85 exir_ops.edge.aten.max_pool2d_with_indices.default,
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A D_qnnpack_pt2e.py119 (operator.getitem, torch.ops.aten.max_pool2d_with_indices.default, 0)
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dsymbolic_script.cpp1306 …output, indices = torch.max_pool2d_with_indices(self, kernel_size, stride, padding, dilation, ceil…
1312 def max_pool2d_with_indices(self,
1318 …output, indices = torch.max_pool2d_with_indices(self, kernel_size, stride, padding, dilation, ceil…
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_subgraph_rewriter.py938 return torch.ops.aten.max_pool2d_with_indices.default(
944 return torch.ops.aten.max_pool2d_with_indices.default(
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dgraph_helper.cpp231 (nodeKind == Symbol::fromQualString("aten::max_pool2d_with_indices"))) { in createOperator()
233 // Once it's supported, max_pool2d_with_indices should be mapped differently in createOperator()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api428 max_pool2d_with_indices
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dfunctions_hifi.yaml75 - op: max_pool2d_with_indices.out
/aosp_15_r20/external/executorch/kernels/aten/
H A Dfunctions.yaml248 - op: max_pool2d_with_indices.out

12345