| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Pooling.cpp | 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 D | DilatedMaxPool2d.cpp | 18 TORCH_META_FUNC(max_pool2d_with_indices) in TORCH_META_FUNC() argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A 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()
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/partition/ |
| H A D | configs.py | 46 exir_ops.edge.aten.max_pool2d_with_indices.default,
95 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | _qnnpack_pt2e.py | 119 (operator.getitem, torch.ops.aten.max_pool2d_with_indices.default, 0)
|
| /aosp_15_r20/external/executorch/backends/arm/operator_support/ |
| H A D | tosa_supported_operators.py | 93 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | alexnet_training.txt | 27 Operator: aten.max_pool2d_with_indices.default
|
| H A D | maml_omniglot_training.txt | 18 Operator: aten.max_pool2d_with_indices.default
|
| H A D | vgg16_training.txt | 35 Operator: aten.max_pool2d_with_indices.default
|
| H A D | resnet18_training.txt | 45 Operator: aten.max_pool2d_with_indices.default
|
| H A D | squeezenet1_1_training.txt | 57 Operator: aten.max_pool2d_with_indices.default
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Pool.cpp | 207 VK_REGISTER_OP(aten.max_pool2d_with_indices.default, max_pool2d);
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | inductor_prims.py | 108 vals, indices = torch.ops.aten.max_pool2d_with_indices(
|
| H A D | decomposition.py | 932 @register_decomposition(aten.max_pool2d_with_indices)
933 def max_pool2d_with_indices( function
|
| /aosp_15_r20/external/executorch/backends/vulkan/partitioner/ |
| H A D | supported_ops.py | 85 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 4063 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/ao/quantization/pt2e/representation/ |
| H A D | rewrite.py | 553 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/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/timm_train/ |
| H A D | resnet18_training.txt | 50 Operator: aten.max_pool2d_with_indices.default
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_matcher_utils.py | 129 return torch.ops.aten.max_pool2d_with_indices.default(
|
| H A D | test_subgraph_rewriter.py | 938 return torch.ops.aten.max_pool2d_with_indices.default(
944 return torch.ops.aten.max_pool2d_with_indices.default(
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A 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];
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | layout_transform.py | 36 exir_ops.edge.aten.max_pool2d_with_indices.default,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | AdaptivePooling.mm | 18 #include <ATen/ops/max_pool2d_with_indices.h>
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | pooling.cpp | 112 return F::detail::max_pool2d_with_indices( in forward_with_indices()
|