mxnet
node.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  */
34 // Acknowledgement: This file originates from incubator-tvm
35 #ifndef MXNET_NODE_NODE_H_
36 #define MXNET_NODE_NODE_H_
37 
39 #include <mxnet/runtime/object.h>
40 #include <mxnet/runtime/memory.h>
41 
42 #include <string>
43 #include <vector>
44 #include <utility>
45 #include <type_traits>
46 
47 namespace mxnet {
48 
49 using runtime::TypeIndex;
50 using runtime::Object;
51 // We strictly restrict ObjectPtr to ::mxnet::runtime
52 // as it may conflict with ::nnvm::ObjectPtr
53 // using runtime::ObjectPtr;
54 using runtime::ObjectRef;
55 using runtime::GetRef;
56 using runtime::Downcast;
57 using runtime::ObjectHash;
58 using runtime::ObjectEqual;
60 
61 } // namespace mxnet
62 
63 #endif // MXNET_NODE_NODE_H_
Runtime memory management.
namespace of mxnet
Definition: api_registry.h:33
ObjectPtr< T > make_object(Args &&...args)
Allocate an object using default allocator.
Definition: memory.h:203
SubRef Downcast(BaseRef ref)
Downcast a base reference type to a more specific type.
Definition: object.h:809
A managed object in MXNet runtime.
TypeIndex
list of the type index.
Definition: object.h:51
RefType GetRef(const ObjectType *ptr)
Get a reference type from a raw object ptr type.