Home
last modified time | relevance | path

Searched full:conv2d_transpose_clamp_prepack (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DRegisterOpContextClass.cpp81 …m.def(TORCH_SELECTIVE_SCHEMA("prepacked::conv2d_transpose_clamp_prepack(Tensor W, Tensor? B, int[2… in TORCH_LIBRARY()
90 …m.impl(TORCH_SELECTIVE_NAME("prepacked::conv2d_transpose_clamp_prepack"), TORCH_FN(createConv2dTra… in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/
H A Dtest_xnnpack_integration.py221 packed_weight_bias = torch.ops.prepacked.conv2d_transpose_clamp_prepack(
502 torch.ops.prepacked.conv2d_transpose_clamp_prepack(
947 "prepacked::conv2d_transpose_clamp_prepack": 1,
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dxnnpack_rewrite.cpp154 %packed_weight_bias = prepacked::conv2d_transpose_clamp_prepack( in insertPrePackedConv2dOp()
408 "prepacked::conv2d_transpose_clamp_prepack")); in FoldPrePackingOps()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_optimize_for_mobile_preserve_debug_info.py137 "prepacked::conv2d_transpose_clamp_prepack": "aten::conv_transpose2d",
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmodel_ops.yaml396 prepacked::conv2d_transpose_clamp_prepack: 1
H A Dcoverage.yaml1077 prepacked::conv2d_transpose_clamp_prepack: 1