File tree Expand file tree Collapse file tree
docs/src/api/NN_Primitives Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -24,7 +24,13 @@ make_causal_mask
2424
2525``` @docs
2626softmax
27+ softmax!
2728logsoftmax
29+ logsoftmax!
30+ ∇softmax
31+ ∇softmax!
32+ ∇logsoftmax
33+ ∇logsoftmax!
2834```
2935
3036## Pooling
@@ -165,22 +171,18 @@ NNlib.is_strided
165171NNlib.conv_direct!
166172NNlib.gemm!
167173NNlib.calc_padding_regions
168- NNlib.∇depthwiseconv_data_im2col!
169174NNlib._prepare_imrotate
170175NNlib.insert_singleton_spatial_dimension
171176NNlib._fast_broadcast!
172177NNlib.hann_window
173178NNlib._rng_from_array
174- NNlib.∇depthwiseconv_filter_im2col!
175179NNlib.istft
176180NNlib.transpose_swapbatch
177181NNlib.transpose_pad
178182NNlib.power_to_db
179183NNlib.col2im!
180- NNlib.depthwiseconv_im2col!
181184NNlib.storage_type
182185NNlib.im2col_dims
183- NNlib.∇depthwiseconv_filter_direct!
184186NNlib.reverse_indices
185187NNlib.∇conv_filter_im2col!
186188NNlib.conv_im2col!
@@ -192,7 +194,6 @@ NNlib.add_blanks
192194NNlib.∇filter_im2col_dims
193195NNlib._bilinear_helper
194196NNlib._triangular_filterbanks
195- NNlib.∇depthwiseconv_data_direct!
196197NNlib.db_to_power
197198NNlib.predilated_size
198199NNlib.stft
@@ -202,7 +203,6 @@ NNlib.BatchedTranspose
202203NNlib._rotate_coordinates
203204NNlib.melscale_filterbanks
204205NNlib.logaddexp
205- NNlib.depthwiseconv_direct!
206206NNlib.im2col!
207207NNlib.predilate
208208NNlib.safe_div
You can’t perform that action at this time.
0 commit comments