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.

Classes
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 >

Namespaces
	mxnet
	namespace of mxnet
	mxnet::runtime
	mxnet::runtime::detail

Macros
#define	MXNET_CHECK_TYPE_CODE(CODE, T)
	convert a string to TVM type. More...

Functions
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>
R	mxnet::runtime::detail::call_packed (const PackedFunc &pf, Args &&...args)

Detailed Description

Type-erased function used across MXNET API.

Macro Definition Documentation

MXNET_CHECK_TYPE_CODE

#define MXNET_CHECK_TYPE_CODE	(		CODE,
			T
	)

Value:

CHECK_EQ(CODE, T) << " expected "                            \
  << TypeCode2Str(T) << " but get " << TypeCode2Str(CODE)      \

convert a string to TVM type.

Parameters

s	The string to be converted.

Returns

The corresponding tvm type.