| 1407 | #ifdef DLIB_USE_CUDA |
| 1408 | |
| 1409 | void test_scale_channels() |
| 1410 | { |
| 1411 | tt::tensor_rand rnd; |
| 1412 | |
| 1413 | resizable_tensor dest1(2,3,4,5), dest2; |
| 1414 | rnd.fill_gaussian(dest1); |
| 1415 | dest2 = dest1; |
| 1416 | |
| 1417 | resizable_tensor src(2,3,4,5); |
| 1418 | resizable_tensor scales(2,3); |
| 1419 | rnd.fill_gaussian(src); |
| 1420 | rnd.fill_gaussian(scales); |
| 1421 | |
| 1422 | |
| 1423 | cpu::scale_channels(true, dest1, src, scales); |
| 1424 | cuda::scale_channels(true, dest2, src, scales); |
| 1425 | |
| 1426 | DLIB_TEST(max(abs(mat(dest1)-mat(dest2))) < 1e-6); |
| 1427 | |
| 1428 | cpu::scale_channels(false, dest1, src, scales); |
| 1429 | cuda::scale_channels(false, dest2, src, scales); |
| 1430 | |
| 1431 | DLIB_TEST(max(abs(mat(dest1)-mat(dest2))) < 1e-6); |
| 1432 | } |
| 1433 | |
| 1434 | // ---------------------------------------------------------------------------------------- |
| 1435 |
no test coverage detected