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 
27 #ifndef MSHADOW_EXTENSION_H_
28 #define MSHADOW_EXTENSION_H_
29 #include "./expr_engine-inl.h"
30 #include "./extension/broadcast.h"
33 #include "./extension/reshape.h"
34 #include "./extension/swapaxis.h"
35 #include "./extension/reduceto1d.h"
40 #include "./extension/pad.h"
41 #include "./extension/crop.h"
42 #include "./extension/mirror.h"
43 #include "./extension/concat.h"
45 #include "./extension/choose.h"
46 #include "./extension/fill.h"
47 #include "./extension/one_hot.h"
48 #include "./extension/slice.h"
49 #include "./extension/slice_ex.h"
50 #include "./extension/take.h"
51 #include "./extension/take_grad.h"
55 #include "./extension/transpose.h"
56 #include "./extension/flip.h"
57 #include "./extension/complex.h"
58 #include "./extension/range.h"
59 #include "./extension/mask.h"
60 #endif // MSHADOW_EXTENSION_H_
support for chpool
support for implicit array filling operation
support for complex operations
support for reshape
support for transpose
support for unpack
support for mirror
support for implicit GEMM operation
support for pad
support for crop
support for implicit array selection operation
support for unpool
support generating a range vector
definitions of how expressions should be evaluated
support for concatenation
support for sum_rows and sumall_except_dim
support for pack
Create one-hot indicator array based on the index.
support for spatial pooling
support for swapaxis
support for slice a certain dimension.
support for broadcast and repmat
support for flip a certain dimension.