Exemplo n.º 1
0
def test_d_on_cuda_sigmoid():
    _test_template(version='D', approx_func=sigmoid, cuda=CUDA)
Exemplo n.º 2
0
def test_d_on_cuda_tanh():
    _test_template(version='D', approx_func=tanh, cuda=CUDA)
Exemplo n.º 3
0
def test_d_on_cuda_lrelu():
    _test_template(version='D', approx_func=leaky_relu, cuda=CUDA)
Exemplo n.º 4
0
def test_c_on_cpu_sigmoid():
    _test_template(version='C', approx_func=sigmoid, cuda=CUDA)
Exemplo n.º 5
0
def test_c_on_cpu_tanh():
    _test_template(version='C', approx_func=tanh, cuda=CUDA)
Exemplo n.º 6
0
def test_c_on_cpu_lrelu():
    _test_template(version='C', approx_func=leaky_relu, cuda=CUDA)