Home
last modified time | relevance | path

Searched refs:test_max_pool2d (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/
H A DMetalOpTestRunner.mm42 REG_TEST("test_max_pool2d", test_max_pool2d);
H A DMPSCNNTests.h8 bool test_max_pool2d();
H A DMPSCNNTests.mm225 bool test_max_pool2d() { function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_quantized_api_test.cpp2415 void test_max_pool2d( in test_max_pool2d() function
2451 test_max_pool2d({1, 3, 72, 96}, c10::ScalarType::QInt8); in TEST_F()
2452 test_max_pool2d({5, 13, 55, 68}, c10::ScalarType::QInt8); in TEST_F()
2457 test_max_pool2d({5, 13, 55, 68}, c10::ScalarType::QUInt8); in TEST_F()
2458 test_max_pool2d({5, 13, 55, 19}, c10::ScalarType::QUInt8); in TEST_F()
/aosp_15_r20/external/pytorch/test/
H A Dtest_nnapi.py423 def test_max_pool2d(self): member in TestNNAPI
H A Dtest_jit_llga_fuser.py193 def test_max_pool2d(self, dtype): member in TestOp
H A Dtest_mkldnn.py736 def test_max_pool2d(self): member in TestMkldnn
H A Dtest_mps.py2236 def test_max_pool2d(self): member in TestMPS
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
H A Dtf-to-tosa-pipeline.mlir457 // CHECK-LABEL: test_max_pool2d
459 func.func @test_max_pool2d(%arg0: tensor<1x32x32x8xf32>) -> tensor<1x32x32x8xf32> {
H A Dtfl-to-tosa-pipeline.mlir635 // CHECK-LABEL: test_max_pool2d
637 func.func @test_max_pool2d(%arg0: tensor<1x32x32x8xf32>) -> tensor<*xf32> {
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp2850 void test_max_pool2d( in test_max_pool2d() function
2933 test_max_pool2d( in TEST()
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_pooling.py1053 def test_max_pool2d(self, device): member in TestPoolingNNDeviceType
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py1441 def test_max_pool2d(self, X, kernel, stride, dilation, padding, ceil_mode): member in TestQuantizedOps