Zero padding test for GEMM kernels */
| 98 | |
| 99 | /* Zero padding test for GEMM kernels */ |
| 100 | bool validate_gemm_zero_padding(const TensorShape shape0, const TensorShape shape1) |
| 101 | { |
| 102 | // Create tensors |
| 103 | TensorInfo in0(shape0, 1, DataType::F32); |
| 104 | TensorInfo in1(shape1, 1, DataType::F32); |
| 105 | TensorInfo dst; |
| 106 | |
| 107 | // Validate zero-padding |
| 108 | cpu::kernels::CpuGemmMatrixMultiplyKernel gemm; |
| 109 | gemm.configure(&in0, &in1, &dst, 1.0, false); |
| 110 | |
| 111 | return in0.padding().empty() && in1.padding().empty() && dst.padding().empty(); |
| 112 | } |
| 113 | } // namespace |
| 114 | |
| 115 | TEST_SUITE(NEON) |