Function test_simple_arithm

dali/test/python/test_gpu_python_function_operator.py:67–68

()

def test_simple_arithm():
    validate_cpu_vs_gpu(arrays_arithmetic, arrays_arithmetic, num_outputs=2)


square_diff_kernel = cupy.ElementwiseKernel("T x, T y", "T z", "z = x*x - y*y", "square_diff")