| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | conv.h | 211 inline Tensor conv_transpose2d( in conv_transpose2d() function
220 return torch::conv_transpose2d( in conv_transpose2d()
227 /// https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.conv_transpose2d
237 /// F::conv_transpose2d(x, weight, F::ConvTranspose2dFuncOptions().stride(1));
239 inline Tensor conv_transpose2d(
243 return detail::conv_transpose2d(
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_optimize_for_mobile_preserve_debug_info.py | 102 return F.conv_transpose2d(
137 "prepacked::conv2d_transpose_clamp_prepack": "aten::conv_transpose2d",
138 "prepacked::conv2d_transpose_clamp_run": "aten::conv_transpose2d",
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | graph_rewrite_helper.cpp | 130 … %r = aten::conv_transpose2d(%a, %w, %b, %stride, %padding, %output_padding, %groups, %dilation) in replaceConvolutionWithAtenConv()
133 std::string conv_transpose2d = R"( in replaceConvolutionWithAtenConv() local
137 … %r = aten::conv_transpose2d(%a, %w, %b, %stride, %padding, %output_padding, %groups, %dilation) in replaceConvolutionWithAtenConv()
252 convolution, conv_transpose2d); in replaceConvolutionWithAtenConv()
|
| H A D | vulkan_rewrite.cpp | 102 …%res = aten::conv_transpose2d(%input, %weight, %bias, %stride, %padding, %output_padding, %groups,… in insertPrePackedConv2dOp()
237 // quantized::conv_transpose2d in rewriteQuantizedOps()
240 %res = quantized::conv_transpose2d(%a_quant, %packed_params, %r_scale, %r_zero_point) in rewriteQuantizedOps()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 434 // aten::conv_transpose2d in quant_fusion_pattern_and_replacements()
435 std::string conv_transpose2d = R"( in quant_fusion_pattern_and_replacements() local
440 …%r = aten::conv_transpose2d(%a_dequant, %w_dequant, %b, %stride, %padding, %output_padding, %group… in quant_fusion_pattern_and_replacements()
447 %r_quant = quantized::conv_transpose2d(%a_quant, %packed_params, %r_scale, %r_zero_point) in quant_fusion_pattern_and_replacements()
959 {"quantized::conv_transpose2d", in quant_fusion_pattern_and_replacements()
960 std::move(conv_transpose2d), in quant_fusion_pattern_and_replacements()
1242 …%r = aten::conv_transpose2d(%a_dequant, %w_dequant, %b, %stride, %padding, %output_padding, %group… in conv_prepack_unpack_patterns()
1250 …%r = aten::conv_transpose2d(%a_dequant, %w_dequant, %b_unpacked, %stride, %padding, %output_paddin… in conv_prepack_unpack_patterns()
|
| H A D | helper.cpp | 40 "conv_transpose2d",
287 {"conv_transpose2d", 1}, in isWeight()
305 {"conv_transpose2d", 2}, in isBiasOfConvOrLinear()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/ |
| H A D | conv_transpose2d.yaml | 7 conv_transpose2d:
17 - NAME: conv_transpose2d
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | amp.rst | 162 ``conv_transpose2d``,
296 ``conv_transpose2d``,
393 ``conv_transpose2d``,
|
| /aosp_15_r20/external/executorch/devtools/size_analysis_tool/ |
| H A D | size_analysis_tool_test.py | 39 self.conv_transpose2d = torch.nn.ConvTranspose2d(
47 x = self.conv_transpose2d(x)
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 149 - aten::conv_transpose2d.input
683 - quantized::conv_transpose2d
780 aten::conv_transpose2d.input: 5
1034 quantized::conv_transpose2d: 2
|
| H A D | model_ops.yaml | 102 aten::conv_transpose2d.input: 8
427 quantized::conv_transpose2d: 1
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | utils.py | 203 torch.ops.aten.conv_transpose2d,
204 torch.ops.aten.conv_transpose2d.input,
216 return conv_fn in [F.conv_transpose1d, F.conv_transpose2d]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 211 ref_result = F.conv_transpose2d(
478 return F.conv_transpose2d(
923 return F.conv_transpose2d(
946 "Tensor = aten::conv_transpose2d": -1,
|
| H A D | test_flop_counter.py | 224 assert_equivalence(lambda: F.conv_transpose2d(x, weight).sum().backward(), 32)
245 lambda: F.conv_transpose2d(x, transposed_weight).sum().backward()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | efficient_conv_bn_eval.py | 173 torch.conv_transpose2d,
264 torch.ops.aten.conv_transpose2d.input,
|
| /aosp_15_r20/external/libopus/dnn/torch/dnntools/dnntools/relegance/ |
| H A D | relegance.py | 412 id_relevance = F.conv_transpose2d(padded_od_relevance, kernel, stride=stride)
415 id_relevance = F.conv_transpose2d(padded_od_relevance, kernel, stride=stride)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.cpp | 221 KERNEL_MPS2(conv_transpose2d, input, lower_precision_fp) in TORCH_LIBRARY_IMPL()
346 KERNEL_CPU(conv_transpose2d, input, lower_precision_fp) in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | conv.h | 393 /// `torch::nn::functional::conv_transpose2d`.
398 /// F::conv_transpose2d(x, weight, F::ConvTranspose2dFuncOptions().stride(1));
|
| /aosp_15_r20/external/executorch/exir/dialects/edge/spec/ |
| H A D | gen.py | 141 opdb_key = "conv_transpose2d"
193 elif opdb_key == "conv_transpose2d":
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset10.py | 1111 output = opset9.conv_transpose2d(
1118 @_onnx_symbolic("quantized::conv_transpose2d")
1137 output = opset9.conv_transpose2d(
|
| H A D | symbolic_opset13.py | 1053 output = opset9.conv_transpose2d(
1060 @_onnx_symbolic("quantized::conv_transpose2d")
1081 output = opset9.conv_transpose2d(
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_init.py | 65 "conv_transpose2d",
66 "conv_transpose2d",
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_mkldnn_pattern_matcher.py | 454 self.conv_transpose2d = torch.nn.ConvTranspose2d(
460 x = self.conv_transpose2d(x)
2659 # We don't support conv_transpose2d for now.
2663 self.conv_transpose2d = torch.nn.ConvTranspose2d(
2668 return self.conv_transpose2d(x)
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 311 ("nn.functional.conv_transpose2d", 201),
421 ("nn.ConvTranspose2d", 57524, "nn.functional.conv_transpose2d"),
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | _lower_to_native_backend.py | 342 F.conv_transpose2d: (torch.ops.quantized.conv_transpose2d, None),
394 F.conv_transpose2d,
|