| 33 | |
| 34 | template<typename T> |
| 35 | void testPad(const vector<T>& input, const dim4& inDims, const dim4& lbPadding, |
| 36 | const dim4& ubPadding, const af::borderType btype, |
| 37 | const vector<T>& gold, const dim4& outDims) { |
| 38 | SUPPORTED_TYPE_CHECK(T); |
| 39 | array in(inDims, input.data()); |
| 40 | array out = af::pad(in, lbPadding, ubPadding, btype); |
| 41 | ASSERT_VEC_ARRAY_EQ(gold, outDims, out); |
| 42 | } |
| 43 | |
| 44 | TYPED_TEST(PadBorders, Zero) { |
| 45 | testPad(vector<TypeParam>({ |