File tree Expand file tree Collapse file tree 1 file changed +11
-1
lines changed
Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change @@ -110,6 +110,7 @@ categorical_crossentropy(
110110 axis=-1
111111)
112112ceil(x)
113+ celu(x, alpha=1.0)
113114cholesky(x)
114115clip(
115116 x,
@@ -223,7 +224,8 @@ dot_product_attention(
223224 bias=None,
224225 mask=None,
225226 scale=None,
226- is_causal=False
227+ is_causal=False,
228+ flash_attention=None
227229)
228230dtype(x)
229231eig(x)
@@ -235,6 +237,7 @@ equal(x1, x2)
235237erf(x)
236238erfinv(x)
237239exp(x)
240+ exp2(x)
238241expand_dims(x, axis)
239242expm1(x)
240243extract_sequences(
@@ -271,11 +274,14 @@ full_like(
271274)
272275gelu(x, approximate=True)
273276get_item(x, key)
277+ glu(x, axis=-1)
274278greater(x1, x2)
275279greater_equal(x1, x2)
280+ hard_shrink(x, threshold=0.5)
276281hard_sigmoid(x)
277282hard_silu(x)
278283hard_swish(x)
284+ hard_tanh(x)
279285histogram(
280286 x,
281287 bins=10,
@@ -290,6 +296,7 @@ in_top_k(
290296 predictions,
291297 k
292298)
299+ inner(x1, x2)
293300inv(x)
294301irfft(x, fft_length=None)
295302is_tensor(x)
@@ -562,6 +569,7 @@ slice_update(
562569 updates
563570)
564571slogdet(x)
572+ soft_shrink(x, threshold=0.5)
565573softmax(x, axis=-1)
566574softplus(x)
567575softsign(x)
@@ -585,6 +593,7 @@ split(
585593)
586594sqrt(x)
587595square(x)
596+ squareplus(x, b=4)
588597squeeze(x, axis=None)
589598stack(x, axis=0)
590599std(
@@ -635,6 +644,7 @@ take_along_axis(
635644)
636645tan(x)
637646tanh(x)
647+ tanh_shrink(x)
638648tensordot(
639649 x1,
640650 x2,
You can’t perform that action at this time.
0 commit comments