Home
last modified time | relevance | path

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

/external/pytorch/test/
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
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 …]
Dtest_mobile_optimizer.py113 .check_not("prepacked::conv2d_clamp_prepack") \
125 .check_not("prepacked::conv2d_clamp_prepack") \
/external/pytorch/torch/csrc/jit/passes/
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()
Dconstant_propagation.h11 // and prepacked::conv2d_clamp_prepack)
/external/pytorch/test/jit/
Dtest_optimize_for_mobile_preserve_debug_info.py135 "prepacked::conv2d_clamp_prepack": "aten::conv2d",
227 "prepacked::conv2d_clamp_prepack": "prepacked::conv2d_clamp_prepack",
/external/pytorch/aten/src/ATen/native/vulkan/ops/
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()
DConvolution.h221 c10::intrusive_ptr<Conv2dOpContext> conv2d_clamp_prepack(
DConvolution.cpp1299 c10::intrusive_ptr<Conv2dOpContext> conv2d_clamp_prepack( in conv2d_clamp_prepack() function
/external/pytorch/aten/src/ATen/native/xnnpack/
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()
/external/pytorch/test/mobile/model_test/
Dmodel_ops.yaml394 prepacked::conv2d_clamp_prepack: 2
Dcoverage.yaml1076 prepacked::conv2d_clamp_prepack: 2
/external/pytorch/test/cpp/tensorexpr/
Dtest_external_calls.cpp581 .findSchemaOrThrow("prepacked::conv2d_clamp_prepack", "") in TEST()
/external/pytorch/aten/src/ATen/test/
Dvulkan_api_test.cpp1535 "vulkan_prepack::conv2d_clamp_prepack", in test_backwards_compatible_conv2d_context()
2137 "prepacked::conv2d_clamp_prepack", in TEST_F()