Home
last modified time | relevance | path

Searched refs:test_conv2d (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtfl-to-tosa-pipeline-filtered.mlir5 // CHECK-LABEL: test_conv2d
8 func.func @test_conv2d(%arg0: tensor<1x32x32x8xf32>, %arg1: tensor<16x2x2x8xf32>) -> tensor<*xf32> {
Dtf-to-tosa-pipeline.mlir9 // CHECK-LABEL: test_conv2d
14 func.func @test_conv2d(%arg0: tensor<1x32x32x8xf32>, %arg1: tensor<2x2x8x16xf32>) -> tensor<1x32x32…
Dtfl-to-tosa-pipeline.mlir13 // CHECK-LABEL: test_conv2d
16 func.func @test_conv2d(%arg0: tensor<1x32x32x8xf32>, %arg1: tensor<16x2x2x8xf32>) -> tensor<*xf32> {
/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/
DMetalOpTestRunner.mm40 REG_TEST("test_conv2d", test_conv2d);
DMPSCNNTests.h6 bool test_conv2d();
DMPSCNNTests.mm162 bool test_conv2d() { function
/external/mesa3d/src/gallium/targets/teflon/
Dtest_model_generation.py200 def test_conv2d(batch_size, input_size, weight_size, in_ch, out_ch, stride, padding, signed, seed): function
218 test_conv2d(1, 80, 5, 16, 128, 2, "same", False, 4)
/external/pytorch/test/mobile/
Dtest_quantize_fx_lite_script_module.py51 def test_conv2d(self): member in TestLiteFuseFx
/external/executorch/backends/arm/test/ops/
Dtest_depthwise_conv.py16 from executorch.backends.arm.test.ops.test_conv2d import Conv2d
/external/pytorch/test/quantization/pt2e/
Dtest_representation.py129 def test_conv2d(self): member in TestPT2ERepresentation
Dtest_xnnpack_quantizer.py71 def test_conv2d(self): member in TestXNNPACKQuantizer
Dtest_x86inductor_quantizer.py584 def test_conv2d(self): member in TestQuantizePT2EX86Inductor
/external/pytorch/test/inductor/
Dtest_snode_runtime.py107 def test_conv2d(self): member in ComputeBoundedTests
/external/pytorch/test/quantization/bc/
Dtest_backward_compatibility.py316 def test_conv2d(self): member in TestSerialization
/external/pytorch/test/
Dtest_nnapi.py507 def test_conv2d(self): member in TestNNAPI
Dtest_xnnpack_integration.py99 def test_conv2d( member in TestXNNPACKOps
322 def test_conv2d( member in TestXNNPACKSerDes
Dtest_jit_llga_fuser.py138 def test_conv2d(self, dtype): member in TestOp
Dtest_mkldnn.py285 def test_conv2d(self): member in TestMkldnn
Dtest_jit_fuser_te.py2168 def test_conv2d(self): member in TestTEFuser
/external/executorch/backends/vulkan/test/
Dvulkan_compute_api_test.cpp2939 void test_conv2d( in test_conv2d() function
2986 test_conv2d( in TEST()
2994 test_conv2d( in TEST()
/external/pytorch/aten/src/ATen/test/
Dvulkan_quantized_api_test.cpp1456 void test_conv2d(bool bias_quantized) { in test_conv2d() function
1561 test_conv2d(false); in TEST_F()
1562 test_conv2d(true); in TEST_F()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py425 def test_conv2d(self): member in NNTest
/external/pytorch/test/functorch/
Dtest_vmap.py3254 def test_conv2d(self): member in TestVmapOperators