| /external/executorch/backends/xnnpack/_passes/ |
| D | remove_getitem_op.py | 15 …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"
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Pooling.cpp | 20 #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()
|
| D | ts_native_functions.yaml | 77 - max_pool2d_with_indices
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesPooling.cpp | 53 …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()
|
| /external/executorch/backends/xnnpack/test/ops/ |
| D | maxpool2d.py | 51 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})
|
| /external/executorch/backends/xnnpack/partition/ |
| D | configs.py | 46 exir_ops.edge.aten.max_pool2d_with_indices.default,
95 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset7.py | 32 "max_pool2d_with_indices",
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_max_pool2d_with_indices.cpp | 74 in_type, ctx, "max_pool2d_with_indices.out", CTYPE, [&]() { in max_pool2d_with_indices_out()
|
| /external/executorch/backends/qualcomm/builders/ |
| D | op_max_pool2d.py | 21 target = ["aten.max_pool2d_with_indices.default"]
|
| /external/executorch/backends/apple/mps/operators/ |
| D | pooling_ops.py | 24 target = "aten.max_pool2d_with_indices.default"
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | pooling.h | 262 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()
|
| /external/executorch/backends/arm/operator_support/ |
| D | tosa_supported_operators.py | 93 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | _qnnpack_pt2e.py | 119 (operator.getitem, torch.ops.aten.max_pool2d_with_indices.default, 0)
|
| /external/pytorch/torchgen/aoti/ |
| D | fallback_ops.py | 79 "aten.max_pool2d_with_indices.default",
|
| /external/executorch/backends/vulkan/partitioner/ |
| D | supported_ops.py | 85 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /external/executorch/backends/cadence/aot/ |
| D | functions_hifi.yaml | 75 - op: max_pool2d_with_indices.out
|
| D | functions.yaml | 85 - op: max_pool2d_with_indices.out
|
| /external/pytorch/functorch/op_analysis/ |
| D | public_api | 428 max_pool2d_with_indices
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | Pool.cpp | 207 VK_REGISTER_OP(aten.max_pool2d_with_indices.default, max_pool2d);
|
| /external/executorch/kernels/aten/ |
| D | functions.yaml | 248 - op: max_pool2d_with_indices.out
|
| /external/pytorch/torch/_inductor/ |
| D | inductor_prims.py | 108 vals, indices = torch.ops.aten.max_pool2d_with_indices(
|
| /external/pytorch/docs/source/ |
| D | export.rst | 198 max_pool2d_with_indices = torch.ops.aten.max_pool2d_with_indices.default(
201 getitem: f32[1, 16, 85, 85] = max_pool2d_with_indices[0];
|
| /external/pytorch/test/edge/ |
| D | selected_operators.yaml | 236 aten::max_pool2d_with_indices.out:
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | AdaptivePooling.mm | 18 #include <ATen/ops/max_pool2d_with_indices.h>
|
| /external/pytorch/test/fx/ |
| D | test_matcher_utils.py | 129 return torch.ops.aten.max_pool2d_with_indices.default(
|