Function check_slice_vs_numpy

dali/test/python/operator_1/test_slice.py:546–556

(device, batch_size, axes, axis_names)

def check_slice_vs_numpy(device, batch_size, axes, axis_names):
    eii_args = [
        SliceArgsIterator(batch_size, image_layout="HWC", axes=axes, axis_names=axis_names)
        for k in range(2)
    ]
    compare_pipelines(
        SlicePipeline(device, batch_size, iter(eii_args[0]), axes=axes, axis_names=axis_names),
        SlicePythonOp(batch_size, iter(eii_args[1]), axes=axes, axis_names=axis_names),
        batch_size=batch_size,
        N_iterations=3,
    )


def test_slice_vs_numpy():