spikingjelly.activation_based.cuda_kernel.auto_cuda package#

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.wrap_return_codes(y, codes)[源代码]#

参数:

y (str | None)
codes (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.float2half2(y, x)[源代码]#

参数:

y (str | None)
x (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.constant(y, x, dtype)[源代码]#

参数:

y (str | None)
x (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.abs(y, x, dtype)[源代码]#

参数:

y (str | None)
x (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.power(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.if_else(z, x, y, mask, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
mask (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.if_else_else(w, x, y, z, mask_x, mask_y, dtype)[源代码]#

参数:

w (str | None)
x (str)
y (str)
z (str)
mask_x (str)
mask_y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.greater_equal(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.greater_than(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.minimal(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.maximum(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.add(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.sub(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.mul(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.div(z, x, y, dtype)[源代码]#

参数:

z (str | None)
x (str)
y (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.neg(y, x, dtype)[源代码]#

参数:

y (str | None)
x (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.heaviside(y, x, dtype)[源代码]#

参数:

y (str | None)
x (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.exp(y, x, dtype)[源代码]#

参数:

y (str | None)
x (str)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.sigmoid(y, x, alpha, dtype)[源代码]#

参数:

y (str | None)
x (str)
alpha (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.sigmoid_backward(y, x, alpha, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.atan_backward(y, x, alpha, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.piecewise_leaky_relu_backward(y, x, w, c, dtype)[源代码]#

参数:

y (str)
x (str)
w (float)
c (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.s2nn_backward(y, x, alpha, beta, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
beta (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.q_pseudo_spike_backward(y, x, alpha, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.leaky_k_relu_backward(y, x, leak, k, dtype)[源代码]#

参数:

y (str)
x (str)
leak (float)
k (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.fake_numerical_gradient_backward(y, x, alpha, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
dtype (str)

spikingjelly.activation_based.cuda_kernel.auto_cuda.cfunction.log_tailed_relu_backward(y, x, alpha, dtype)[源代码]#

参数:

y (str)
x (str)
alpha (float)
dtype (str)