mxnet
cpp-package
include
mxnet-cpp
base.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 MXNET_CPP_BASE_H_
27
#define MXNET_CPP_BASE_H_
28
29
#include <cstdlib>
30
#include "
mxnet/c_api.h
"
31
#include "
nnvm/c_api.h
"
32
33
namespace
mxnet
{
34
namespace
cpp {
35
36
typedef
unsigned
index_t
;
37
38
enum
OpReqType
{
40
kNullOp
,
42
kWriteTo
,
48
kWriteInplace
,
50
kAddTo
51
};
52
53
}
// namespace cpp
54
}
// namespace mxnet
55
56
#endif // MXNET_CPP_BASE_H_
mxnet
namespace of mxnet
Definition:
api_registry.h:33
mxnet::cpp::kNullOp
@ kNullOp
no operation, do not write anything
Definition:
base.h:40
mxnet::cpp::kAddTo
@ kAddTo
add to the provided space
Definition:
base.h:50
mxnet::cpp::index_t
unsigned index_t
Definition:
base.h:36
c_api.h
C API of mxnet.
c_api.h
C API of NNVM symbolic construction and pass. Enables construction and transformation of Graph in any...
mxnet::cpp::OpReqType
OpReqType
Definition:
base.h:38
mxnet::cpp::kWriteTo
@ kWriteTo
write gradient to provided space
Definition:
base.h:42
mxnet::cpp::kWriteInplace
@ kWriteInplace
perform an inplace write, Target shares memory with one of input arguments. This option only happen w...
Definition:
base.h:48
Generated on Thu Jan 5 2023 03:47:39 for mxnet by
1.8.17