mxnet
extension.h
Go to the documentation of this file.
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  */
19 
26 #ifndef MSHADOW_EXTENSION_H_
27 #define MSHADOW_EXTENSION_H_
28 #include "./expr_engine-inl.h"
29 #include "./extension/broadcast.h"
32 #include "./extension/reshape.h"
33 #include "./extension/swapaxis.h"
34 #include "./extension/reduceto1d.h"
39 #include "./extension/pad.h"
40 #include "./extension/crop.h"
41 #include "./extension/mirror.h"
42 #include "./extension/concat.h"
44 #include "./extension/choose.h"
45 #include "./extension/fill.h"
46 #include "./extension/one_hot.h"
47 #include "./extension/slice.h"
48 #include "./extension/slice_ex.h"
49 #include "./extension/take.h"
50 #include "./extension/take_grad.h"
54 #include "./extension/transpose.h"
55 #include "./extension/flip.h"
56 #include "./extension/complex.h"
57 #include "./extension/range.h"
58 #include "./extension/mask.h"
59 #endif // MSHADOW_EXTENSION_H_
swapaxis.h
support for swapaxis
pad.h
support for pad
unpack_patch2col.h
support for unpack
one_hot.h
Create one-hot indicator array based on the index.
take_grad.h
mirror.h
support for mirror
reduceto1d.h
support for sum_rows and sumall_except_dim
complex.h
support for complex operations
concat.h
support for concatenation
spatial_upsampling_nearest.h
pack_col2patch.h
support for pack
reduce_with_axis.h
take.h
flip.h
support for flip a certain dimension.
mask.h
channel_pool.h
support for chpool
broadcast.h
support for broadcast and repmat
broadcast_with_axis.h
spatial_unpool.h
support for unpool
range.h
support generating a range vector
slice_ex.h
fill.h
support for implicit array filling operation
spatial_pool.h
support for spatial pooling
reshape.h
support for reshape
implicit_gemm.h
support for implicit GEMM operation
choose.h
support for implicit array selection operation
crop.h
support for crop
transpose.h
support for transpose
slice.h
support for slice a certain dimension.
expr_engine-inl.h
definitions of how expressions should be evaluated
channel_unpool.h