Classes | Namespaces | Macros | Functions
packed_func.h File Reference

Type-erased function used across MXNET API. More...

#include <dmlc/logging.h>
#include <mxnet/runtime/c_runtime_api.h>
#include <mxnet/runtime/object.h>
#include <mxnet/runtime/ndarray.h>
#include <mxnet/runtime/container.h>
#include <mxnet/runtime/ffi_helper.h>
#include <mxnet/runtime/data_type.h>
#include <mxnet/node/container.h>
#include <mxnet/ir/expr.h>
#include <mxnet/ndarray.h>
#include <mxnet/base.h>
#include <functional>
#include <tuple>
#include <vector>
#include <string>
#include <limits>
#include <memory>
#include <utility>
#include <type_traits>
#include <sstream>
Include dependency graph for packed_func.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.


class  mxnet::runtime::PackedFunc
 Packed function is a type-erased function. The arguments are passed by packed format. More...
class  mxnet::runtime::TypedPackedFunc< FType >
 Please refer to TypedPackedFunc<R(Args..)>. More...
class  mxnet::runtime::TypedPackedFunc< R(Args...)>
 A PackedFunc wrapper to provide typed function signature. It is backed by a PackedFunc internally. More...
class  mxnet::runtime::MXNetArgs
 Arguments into TVM functions. More...
struct  mxnet::runtime::extension_type_info< T >
 Type traits to mark if a class is tvm extension type. More...
class  mxnet::runtime::MXNetPODValue_
 Internal base class to handle conversion to POD values. More...
class  mxnet::runtime::MXNetArgValue
 A single argument value to PackedFunc. Containing both type_code and MXNetValue. More...
class  mxnet::runtime::MXNetRetValue
 Return Value container, Unlike MXNetArgValue, which only holds reference and do not delete the underlying container during destruction. More...
struct  mxnet::runtime::detail::for_each_dispatcher< stop, I, F >
struct  mxnet::runtime::detail::for_each_dispatcher< true, I, F >
class  mxnet::runtime::MXNetArgsSetter
struct  mxnet::runtime::detail::unpack_call_dispatcher< R, nleft, index, F >
struct  mxnet::runtime::detail::unpack_call_dispatcher< R, 0, index, F >
struct  mxnet::runtime::detail::unpack_call_dispatcher< void, 0, index, F >
struct  mxnet::runtime::detail::typed_packed_call_dispatcher< R >
struct  mxnet::runtime::detail::typed_packed_call_dispatcher< void >
struct  mxnet::runtime::detail::MXNetValueCast< T, TSrc, is_ext, is_nd >


 namespace of mxnet


 convert a string to TVM type. More...


DLDataType mxnet::runtime::String2DLDataType (std::string s)
 convert a string to TVM type. More...
const char * mxnet::runtime::TypeCode2Str (int type_code)
 Convert type code to its name. More...
int mxnet::runtime::String2MXNetTypeWithBool (const std::string &s)
int mxnet::runtime::String2MXNetType (const std::string &s)
std::ostream & mxnet::runtime::operator<< (std::ostream &os, DLDataType t)
std::ostream & mxnet::runtime::operator<< (std::ostream &os, const MXNetDataType &dtype)
template<typename F , typename... Args>
void mxnet::runtime::detail::for_each (const F &f, Args &&...args)
template<typename R , int nargs, typename F >
void mxnet::runtime::detail::unpack_call (const F &f, const MXNetArgs &args, MXNetRetValue *rv)
template<typename R , typename... Args>
mxnet::runtime::detail::call_packed (const PackedFunc &pf, Args &&...args)

Detailed Description

Type-erased function used across MXNET API.

Macro Definition Documentation

CHECK_EQ(CODE, T) << " expected " \
<< TypeCode2Str(T) << " but get " << TypeCode2Str(CODE) \
const char * TypeCode2Str(int type_code)
Convert type code to its name.
Definition: packed_func.h:851

convert a string to TVM type.

sThe string to be converted.
The corresponding tvm type.