Home
last modified time | relevance | path

Searched full:conv2d_clamp_prepack (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_xnnpack_integration.py139 packed_weight_bias = torch.ops.prepacked.conv2d_clamp_prepack(
364 self.packed_weight_bias = torch.ops.prepacked.conv2d_clamp_prepack(
660 torch.ops.prepacked.conv2d_clamp_prepack(
937 "prepacked::conv2d_clamp_prepack": 1,
1003 "prepacked::conv2d_clamp_prepack": 1,
1011 pattern_count_map["prepacked::conv2d_clamp_prepack"] = -1
1021 "prepacked::conv2d_clamp_prepack": -1,
1029 pattern_count_map["prepacked::conv2d_clamp_prepack"] = -1
1043 "prepacked::conv2d_clamp_prepack": -1,
1051 pattern_count_map["prepacked::conv2d_clamp_prepack"] = -1
[all …]
H A Dtest_vulkan.py84 "vulkan_prepack::conv2d_clamp_prepack": 1,
106 "vulkan_prepack::conv2d_clamp_prepack": 1,
112 pattern_count_map["vulkan_prepack::conv2d_clamp_prepack"] = -1
145 "vulkan_prepack::conv2d_clamp_prepack": 1,
149 pattern_count_map["vulkan_prepack::conv2d_clamp_prepack"] = -1
H A Dtest_mobile_optimizer.py113 .check_not("prepacked::conv2d_clamp_prepack") \
125 .check_not("prepacked::conv2d_clamp_prepack") \
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dxnnpack_rewrite.cpp129 %packed_weight_bias = prepacked::conv2d_clamp_prepack( in insertPrePackedConv2dOp()
184 …ed_weight_bias : __torch__.torch.classes.xnnpack.Conv2dOpContext = prepacked::conv2d_clamp_prepack( in fuseHardtanhWithPackedOps()
209 %packed_weight_bias = prepacked::conv2d_clamp_prepack( in fuseHardtanhWithPackedOps()
235 %packed_weight_bias = prepacked::conv2d_clamp_prepack( in fuseHardtanhWithPackedOps()
278 …ed_weight_bias : __torch__.torch.classes.xnnpack.Conv2dOpContext = prepacked::conv2d_clamp_prepack( in fuseReluWithPackedOps()
304 %packed_weight_bias = prepacked::conv2d_clamp_prepack( in fuseReluWithPackedOps()
331 %packed_weight_bias = prepacked::conv2d_clamp_prepack( in fuseReluWithPackedOps()
405 Symbol::fromQualString("prepacked::conv2d_clamp_prepack") || in FoldPrePackingOps()
H A Dconstant_propagation.h11 // and prepacked::conv2d_clamp_prepack)
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DRegister.cpp152 return conv2d_clamp_prepack( in TORCH_LIBRARY()
171 "vulkan_prepack::conv2d_clamp_prepack(Tensor W, Tensor? B, int[2] stride, " in TORCH_LIBRARY()
274 TORCH_SELECTIVE_NAME("vulkan_prepack::conv2d_clamp_prepack"), in TORCH_LIBRARY_IMPL()
275 TORCH_FN(conv2d_clamp_prepack)); // Backwards compatibility in TORCH_LIBRARY_IMPL()
H A DConvolution.h221 c10::intrusive_ptr<Conv2dOpContext> conv2d_clamp_prepack(
H A DConvolution.cpp1299 c10::intrusive_ptr<Conv2dOpContext> conv2d_clamp_prepack( in conv2d_clamp_prepack() function
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_optimize_for_mobile_preserve_debug_info.py135 "prepacked::conv2d_clamp_prepack": "aten::conv2d",
227 "prepacked::conv2d_clamp_prepack": "prepacked::conv2d_clamp_prepack",
/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DRegisterOpContextClass.cpp80 …m.def(TORCH_SELECTIVE_SCHEMA("prepacked::conv2d_clamp_prepack(Tensor W, Tensor? B, int[2] stride, … in TORCH_LIBRARY()
89 …m.impl(TORCH_SELECTIVE_NAME("prepacked::conv2d_clamp_prepack"), TORCH_FN(createConv2dClampPrePackO… in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmodel_ops.yaml394 prepacked::conv2d_clamp_prepack: 2
H A Dcoverage.yaml1076 prepacked::conv2d_clamp_prepack: 2
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_external_calls.cpp581 .findSchemaOrThrow("prepacked::conv2d_clamp_prepack", "") in TEST()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp1535 "vulkan_prepack::conv2d_clamp_prepack", in test_backwards_compatible_conv2d_context()
2137 "prepacked::conv2d_clamp_prepack", in TEST_F()