| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/pruner/ |
| H A D | prune_functions.py | 172 def _prune_conv2d_helper(conv2d: nn.Conv2d) -> Tensor:
188 def prune_conv2d_padded(conv2d_1: nn.Conv2d) -> None:
221 def prune_conv2d(conv2d: nn.Conv2d) -> None:
227 def prune_conv2d_conv2d(conv2d_1: nn.Conv2d, conv2d_2: nn.Conv2d) -> None:
232 conv2d_1: nn.Conv2d,
234 conv2d_2: nn.Conv2d,
290 c1: nn.Conv2d,
293 c2: nn.Conv2d,
299 c1: nn.Conv2d,
302 c2: nn.Conv2d,
[all …]
|
| /aosp_15_r20/external/executorch/backends/xnnpack/operators/ |
| H A D | op_conv2d.py | 29 class Conv2d(NodeVisitor): class
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_conv2d.py | 29 class Conv2d(NodeVisitor): class
|
| /aosp_15_r20/external/pytorch/test/mobile/lightweight_dispatch/ |
| H A D | test_codegen_unboxing.cpp | 149 TEST(LiteInterpreterTest, Conv2d) { in TEST() argument
|
| /aosp_15_r20/external/executorch/backends/arm/test/ops/ |
| H A D | test_conv2d.py | 19 class Conv2d(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | conv.py | 185 class Conv2d(_ConvNd, nn.Conv2d): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | conv.py | 108 class Conv2d(nnq.Conv2d): class
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ops/ |
| H A D | conv2d.py | 20 class Conv2d(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| H A D | conv.py | 117 class Conv2d(_ConvNd, nn.Conv2d): class
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | conv.py | 311 class Conv2d(_ConvNd): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | conv.py | 483 class Conv2d(_ConvNd): class
|
| /aosp_15_r20/external/ComputeLibrary/ |
| H A D | filelist.json | 270 "Conv2d": { object
1058 "Conv2d": { object
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_modules.py | 3016 class Conv2d(torch.nn.Conv2d): class
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 7207 class Conv2d final : public BaseOp { class
7209 Conv2d( in Conv2d() function in __anon75b2796d0211::Conv2d
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 1017 class Conv2d(torch.nn.Module): class
|