mxnet
|
Namespaces | |
cuda | |
common utils for cuda | |
helper | |
Helper functions. | |
Classes | |
class | LazyAllocArray |
class | ObjectPool |
Object pool for fast allocation and deallocation. More... | |
struct | ObjectPoolAllocatable |
Helper trait class for easy allocation and deallocation. More... | |
Typedefs | |
typedef std::mt19937 | RANDOM_ENGINE |
Random Engine. More... | |
Functions | |
int | GetNumThreadPerGPU () |
int | GetExecNumMatchColor () |
template<class T , class... Args> | |
helper::UniqueIf< T >::SingleObject | MakeUnique (Args &&...args) |
Constructs an object of type T and wraps it in a std :: unique_ptr . More... | |
template<class T > | |
helper::UniqueIf< T >::UnknownBound | MakeUnique (size_t n) |
Constructs an object of type T and wraps it in a std :: unique_ptr . More... | |
template<class T , class... Args> | |
helper::UniqueIf< T >::KnownBound | MakeUnique (Args &&...args)=delete |
Constructs an object of type T and wraps it in a std :: unique_ptr . More... | |
template<typename FCompType > | |
FCompType | GetFCompute (const nnvm::Op *op, const std::string &name, const Context &ctx) |
typedef std::mt19937 mxnet::common::RANDOM_ENGINE |
Random Engine.
|
inline |
FCompType mxnet::common::GetFCompute | ( | const nnvm::Op * | op, |
const std::string & | name, | ||
const Context & | ctx | ||
) |
|
inline |
|
delete |
Constructs an object of type T
and wraps it in a std
::
unique_ptr
.
args | List of arguments with which an instance of T will be constructed. |
std
::
unique_ptr
of an instance of type T
.Constructs a non-array type T
. The arguments args
are passed to the constructor of T
. The function does not participate in the overload resolution if T
is an array type.
helper::UniqueIf<T>::UnknownBound mxnet::common::MakeUnique | ( | size_t | n | ) |
Constructs an object of type T
and wraps it in a std
::
unique_ptr
.
n | The size of the array to construct. |
std
::
unique_ptr
of an instance of type T
.Constructs an array of unknown bound T
. The function does not participate in the overload resolution unless T
is an array of unknown bound.
|
delete |
Constructs an object of type T
and wraps it in a std
::
unique_ptr
.
args | List of arguments with which an instance of T will be constructed. |
Constructs an arrays of known bound is disallowed.
args | List of arguments with which an instance of T will be constructed. |
std
::
unique_ptr
of an instance of type T
.Constructs a non-array type T
. The arguments args
are passed to the constructor of T
. The function does not participate in the overload resolution if T
is an array type.