Home
last modified time | relevance | path

Searched refs:test_avg_pool2d (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_nnapi.py432 def test_avg_pool2d(self): member in TestNNAPI
H A Dtest_jit_llga_fuser.py214 def test_avg_pool2d(self, dtype): member in TestOp
H A Dtest_mkldnn.py859 def test_avg_pool2d(self): member in TestMkldnn
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
H A Dtf-to-tosa-pipeline.mlir448 // CHECK-LABEL: test_avg_pool2d
450 func.func @test_avg_pool2d(%arg0: tensor<1x32x32x8xf32>) -> tensor<1x32x32x8xf32> {
H A Dtfl-to-tosa-pipeline.mlir617 // CHECK-LABEL: test_avg_pool2d
619 func.func @test_avg_pool2d(%arg0: tensor<1x32x32x8xf32>) -> tensor<*xf32> {
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_operators.py411 def test_avg_pool2d(self): member in TestOperators
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py1672 …def test_avg_pool2d(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_overri… member in TestQuantizedOps
7822 def test_avg_pool2d( member in TestQNNPackOps