MCPcopy Create free account
hub / github.com/NVIDIA/DALI / index_pipe

Function index_pipe

dali/test/python/operator_2/test_subscript.py:24–30  ·  view source on GitHub ↗
(data_source, indexing_func)

Source from the content-addressed store, hash-verified

22
23@dali.pipeline_def(batch_size=2, num_threads=3, device_id=0)
24def 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
33def test_plain_indexing():

Callers 14

test_plain_indexingFunction · 0.85
_test_indexingFunction · 0.85
test_swapped_endsFunction · 0.85
test_runtime_indexingFunction · 0.85
test_runtime_stride_dim1Function · 0.85
test_runtime_stride_dim2Function · 0.85
_test_out_of_rangeFunction · 0.85
_test_too_many_indicesFunction · 0.85
test_zero_stride_errorFunction · 0.85
test_empty_sliceFunction · 0.85
test_range_truncationFunction · 0.85

Calls 1

gpuMethod · 0.45

Tested by

no test coverage detected