mxnet
c_api_test.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 
25 #ifndef MXNET_C_API_TEST_H_
26 #define MXNET_C_API_TEST_H_
27 
29 #ifdef __cplusplus
30 extern "C" {
31 #endif // __cplusplus
32 
33 #include <mxnet/c_api.h>
34 
42  const char* prop_name,
43  const mx_uint num_ops,
44  const char** op_names,
45  SymbolHandle* ret_sym_handle);
46 
52 MXNET_DLL int MXSetSubgraphPropertyOpNames(const char* prop_name,
53  const mx_uint num_ops,
54  const char** op_names);
55 
60 MXNET_DLL int MXRemoveSubgraphPropertyOpNames(const char* prop_name);
61 
62 #ifdef __cplusplus
63 }
64 #endif // __cplusplus
65 
66 #endif // MXNET_C_API_TEST_H_
C API of mxnet.
void * SymbolHandle
handle to a symbol that can be bind as operator
Definition: c_api.h:75
MXNET_DLL int MXSetSubgraphPropertyOpNames(const char *prop_name, const mx_uint num_ops, const char **op_names)
Given a subgraph property name, use the provided op names as the op_names attribute for that subgraph...
unsigned int mx_uint
manually define unsigned int
Definition: c_api.h:58
#define MXNET_DLL
MXNET_DLL prefix for windows.
Definition: c_api.h:54
MXNET_DLL int MXPartitionGraphByOpNames(SymbolHandle sym_handle, const char *prop_name, const mx_uint num_ops, const char **op_names, SymbolHandle *ret_sym_handle)
Inhibit C++ name-mangling for MXNet functions.
MXNET_DLL int MXRemoveSubgraphPropertyOpNames(const char *prop_name)
Given a subgraph property name, delete the op name set in the SubgraphPropertyOpNameSet.