Home
last modified time | relevance | path

Searched refs:conv_transpose1d (Results 1 – 25 of 38) sorted by relevance

12

/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dconv.h166 inline Tensor conv_transpose1d( in conv_transpose1d() function
175 return torch::conv_transpose1d( in conv_transpose1d()
194 inline Tensor conv_transpose1d(
198 return detail::conv_transpose1d(
/external/pytorch/torch/csrc/jit/passes/
Dgraph_rewrite_helper.cpp119 std::string conv_transpose1d = R"( in replaceConvolutionWithAtenConv() local
245 convolution, conv_transpose1d); in replaceConvolutionWithAtenConv()
Dautocast.cpp369 case aten::conv_transpose1d: in handleBlock()
/external/pytorch/torch/_inductor/fx_passes/
Defficient_conv_bn_eval.py172 torch.conv_transpose1d,
263 torch.ops.aten.conv_transpose1d.default,
/external/pytorch/torch/ao/quantization/pt2e/
Dutils.py201 torch.ops.aten.conv_transpose1d,
202 torch.ops.aten.conv_transpose1d.default,
216 return conv_fn in [F.conv_transpose1d, F.conv_transpose2d]
Dqat_utils.py645 m, F.conv_transpose1d, _conv1d_bn_example_inputs, is_cuda=is_cuda
851 m, F.conv_transpose1d, _quantized_conv1d_bn_example_inputs, is_cuda=is_cuda
/external/pytorch/docs/source/
Dnn.functional.rst19 conv_transpose1d
Damp.rst161 ``conv_transpose1d``,
295 ``conv_transpose1d``,
392 ``conv_transpose1d``,
/external/pytorch/test/
Dtest_flop_counter.py166 return F.conv_transpose1d(inp, weight)
178 grad_inp = F.conv_transpose1d(grad_out, weight)
197 return F.conv_transpose1d(x, weight).pow(2).sum()
/external/pytorch/aten/src/ATen/
Dautocast_mode.cpp220 KERNEL_MPS(conv_transpose1d, lower_precision_fp) in TORCH_LIBRARY_IMPL()
345 KERNEL_CPU(conv_transpose1d, lower_precision_fp) in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/ao/quantization/
Dqconfig_mapping.py97 .set_object_type(torch.nn.functional.conv_transpose1d, qconfig_transpose)
/external/pytorch/torch/csrc/jit/passes/quantization/
Dquantization_patterns.h419 std::string conv_transpose1d = R"( in quant_fusion_pattern_and_replacements() local
957 std::move(conv_transpose1d), in quant_fusion_pattern_and_replacements()
/external/pytorch/torch/ao/quantization/fx/
D_lower_to_native_backend.py341 F.conv_transpose1d: (torch.ops.quantized.conv_transpose1d, None),
393 F.conv_transpose1d,
895 if func_node.target == F.conv_transpose1d:
Dutils.py161 torch.nn.functional.conv_transpose1d: torch.ops.quantized.conv_transpose1d_prepack,
/external/libopus/dnn/torch/dnntools/dnntools/relegance/
Drelegance.py409 id_relevance = F.conv_transpose1d(padded_od_relevance, kernel, stride=stride)
/external/pytorch/torch/nn/utils/_expanded_weights/
Dconv_utils.py174 func, F.conv_transpose1d, F.conv_transpose2d, F.conv_transpose3d
/external/pytorch/torch/ao/nn/quantized/reference/modules/
Dconv.py333 result = F.conv_transpose1d(
/external/pytorch/test/export/
Dtesting.py183 aten.conv_transpose1d.default,
/external/pytorch/torch/csrc/api/src/nn/modules/
Dconv.cpp261 return F::detail::conv_transpose1d( in forward()
/external/pytorch/test/nn/
Dtest_lazy_modules.py375 torch.nn.functional.conv_transpose1d,
387 torch.nn.functional.conv_transpose1d,
/external/pytorch/functorch/op_analysis/
Dpublic_api452 nn.functional.conv_transpose1d
/external/pytorch/torch/ao/ns/fx/
Dmappings.py148 F.conv_transpose1d,
/external/pytorch/test/cpp_api_parity/
Dparity-tracker.md161 F::conv_transpose1d|Yes|No
/external/pytorch/torch/ao/quantization/backend_config/
D_common_operator_config_utils.py66 F.conv_transpose1d,
/external/pytorch/test/quantization/pt2e/
Dtest_quantize_pt2e_qat.py970 torch.ops.aten.conv_transpose1d,
971 torch.ops.aten.conv_transpose1d.default,

12