(data_source, indexing_func)
| 22 | |
| 23 | @dali.pipeline_def(batch_size=2, num_threads=3, device_id=0) |
| 24 | def index_pipe(data_source, indexing_func): |
| 25 | src = data_source |
| 26 | |
| 27 | cpu = indexing_func(src) |
| 28 | gpu = indexing_func(src.gpu()) |
| 29 | |
| 30 | return src, cpu, gpu |
| 31 | |
| 32 | |
| 33 | def test_plain_indexing(): |
no test coverage detected