mxnet
Classes | Namespaces | Typedefs | Functions
utils.h File Reference

Basic utilility functions. More...

#include <dmlc/logging.h>
#include <dmlc/omp.h>
#include <nnvm/graph.h>
#include <mxnet/engine.h>
#include <mxnet/ndarray.h>
#include <mxnet/op_attr_types.h>
#include <mxnet/graph_attr_types.h>
#include <nnvm/graph_attr_types.h>
#include <memory>
#include <vector>
#include <type_traits>
#include <utility>
#include <random>
#include <string>
#include <thread>
#include <algorithm>
#include <functional>
#include <limits>
#include "../operator/mxnet_op.h"
Include dependency graph for utils.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mxnet::common::csr_indptr_check
 IndPtr should be non-negative, in non-decreasing order, start with 0 and end with value equal with size of indices. More...
 
struct  mxnet::common::csr_idx_check
 Indices should be non-negative, less than the number of columns and in ascending order per row. More...
 
struct  mxnet::common::rsp_idx_check
 Indices of RSPNDArray should be non-negative, less than the size of first dimension and in ascending order. More...
 
struct  mxnet::common::helper::UniqueIf< T >
 Helper for non-array type T. More...
 
struct  mxnet::common::helper::UniqueIf< T[]>
 Helper for an array of unknown bound T. More...
 
struct  mxnet::common::helper::UniqueIf< T[kSize]>
 Helper for an array of known bound T. More...
 

Namespaces

 mxnet
 namespace of mxnet
 
 mxnet::common
 
 mxnet::common::helper
 Helper functions.
 

Typedefs

typedef std::mt19937 mxnet::common::RANDOM_ENGINE
 Random Engine. More...
 

Functions

template<typename xpu >
void mxnet::common::CheckFormatWrapper (const RunContext &rctx, const NDArray &input, const TBlob &err_cpu, const bool full_check)
 
template<typename xpu >
void mxnet::common::CheckFormatCSRImpl (const RunContext &rctx, const NDArray &input, const TBlob &err_cpu, const bool full_check)
 Check the validity of CSRNDArray. More...
 
template<typename xpu >
void mxnet::common::CheckFormatRSPImpl (const RunContext &rctx, const NDArray &input, const TBlob &err_cpu, const bool full_check)
 Check the validity of RowSparseNDArray. More...
 
template<typename xpu >
void mxnet::common::CheckFormatImpl (const RunContext &rctx, const NDArray &input, const TBlob &err_cpu, const bool full_check)
 
template<typename xpu >
void mxnet::common::SparseRetainOpForwardRspWrapper (mshadow::Stream< xpu > *s, const NDArray &input_nd, const TBlob &idx_data, const OpReqType req, NDArray *output_nd)
 Pick rows specified by user input index array from a row sparse ndarray and save them in the output sparse ndarray. More...
 
template<typename xpu >
void mxnet::common::CastStorageDispatch (const OpContext &ctx, const NDArray &input, const NDArray &output)
 
bool mxnet::common::ContainsOnlyStorage (const StorageTypeVector &vstorage, const NDArrayStorageType stype)
 returns true if all storage types in vstorage are the same as target stype. false is returned for empty inputs. More...
 
bool mxnet::common::ContainsOnlyStorage (const StorageTypeVector &vstorage, const NDArrayStorageType stype1, const NDArrayStorageType stype2, bool *has_both)
 returns true if all storage types in vstorage are the same as target stype1 or `stype2'. Sets boolean if both found. false is returned for empty inputs. More...
 
bool mxnet::common::ContainsOnlyStorage (const std::vector< NDArray > &ndarrays, const NDArrayStorageType stype)
 returns true if the storage types of arrays in ndarrays are the same as target stype. false is returned for empty inputs. More...
 
bool mxnet::common::ContainsOnlyStorage (const std::vector< NDArray > &ndarrays, const NDArrayStorageType stype1, const NDArrayStorageType stype2, bool *has_both)
 returns true if the storage types of arrays in ndarrays are the same as targets stype1 or stype2. false is returned for empty inputs. More...
 
bool mxnet::common::ContainsStorageType (const std::vector< NDArray > &ndarrays, const NDArrayStorageType stype)
 returns true if storage type of any array in ndarrays is the same as the target stype. false is returned for empty inputs. More...
 
bool mxnet::common::ContainsStorageType (const std::vector< int > &ndstypes, const NDArrayStorageType stype)
 returns true if any storage type ndstype in ndstypes is the same as the target stype. false is returned for empty inputs. More...
 
std::string mxnet::common::dispatch_mode_string (const DispatchMode x)
 get string representation of dispatch_mode More...
 
std::string mxnet::common::stype_string (const int x)
 get string representation of storage_type More...
 
std::string mxnet::common::dev_type_string (const int dev_type)
 get string representation of device type More...
 
std::string mxnet::common::operator_stype_string (const nnvm::NodeAttrs &attrs, const int dev_mask, const std::vector< int > &in_attrs, const std::vector< int > &out_attrs)
 get string representation of the operator stypes More...
 
std::string mxnet::common::operator_string (const nnvm::NodeAttrs &attrs, const OpContext &ctx, const std::vector< NDArray > &inputs, const std::vector< OpReqType > &req, const std::vector< NDArray > &outputs)
 get string representation of the operator More...
 
void mxnet::common::LogOnce (const std::string &message)
 log message once. Intended for storage fallback warning messages. More...
 
void mxnet::common::LogStorageFallback (const nnvm::NodeAttrs &attrs, const int dev_mask, const std::vector< int > *in_attrs, const std::vector< int > *out_attrs)
 log storage fallback event More...
 
int mxnet::common::GetNumThreadsPerGPU ()
 
int mxnet::common::GetExecNumMatchColor ()
 
template<typename T , typename V >
mxnet::common::ParallelAccumulate (const T *a, const int n, V start)
 
template<typename RandomIt , typename Compare >
void mxnet::common::ParallelSortHelper (RandomIt first, size_t len, size_t grainsize, const Compare &comp)
 Helper function for ParallelSort. DO NOT call this function directly. Use the interface ParallelSort instead. Ref: https://github.com/dmlc/difacto/blob/master/src/common/parallel_sort.h. More...
 
template<typename RandomIt , typename Compare >
void mxnet::common::ParallelSort (RandomIt first, RandomIt last, size_t num_threads, Compare comp)
 Sort the elements in the range [first, last) into the ascending order defined by the comparator comp. If the length of the range [first, last) is greater than a certain threshold, the range will be recursively divided into two and assign two threads to sort each half range. Ref: https://github.com/dmlc/difacto/blob/master/src/common/parallel_sort.h. More...
 
template<typename RandomIt >
void mxnet::common::ParallelSort (RandomIt first, RandomIt last, size_t num_threads)
 Sort the elements in the range [first, last) into ascending order. The elements are compared using the default < operator. If the length of the range [first, last) is greater than a certain threshold, the range will be recursively divided into two and assign two threads to sort each half range. Ref: https://github.com/dmlc/difacto/blob/master/src/common/parallel_sort.h. More...
 
template<class T , class... Args>
helper::UniqueIf< T >::SingleObject mxnet::common::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 mxnet::common::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 mxnet::common::MakeUnique (Args &&...args)=delete
 Constructs an object of type T and wraps it in a std::unique_ptr. More...
 
template<typename FCompType >
FCompType mxnet::common::GetFCompute (const nnvm::Op *op, const std::string &name, const Context &ctx)
 
template<typename T >
constexpr size_t mxnet::common::MaxIntegerValue ()
 Return the max integer value representable in the type T without loss of precision. More...
 
template<>
constexpr size_t mxnet::common::MaxIntegerValue< mshadow::half::half_t > ()
 
MSHADOW_XINLINE int mxnet::common::ilog2ul (size_t a)
 
MSHADOW_XINLINE int mxnet::common::ilog2ui (unsigned int a)
 
NDArray mxnet::common::InitZeros (const NDArrayStorageType stype, const TShape &shape, const Context &ctx, const int dtype)
 Return an NDArray of all zeros. More...
 
void mxnet::common::EmplaceBackZeros (const NDArrayStorageType stype, const TShape &shape, const Context &ctx, const int dtype, std::vector< NDArray > *vec)
 Helper to add a NDArray of zeros to a std::vector. More...
 

Detailed Description

Basic utilility functions.

Copyright (c) 2015 by Contributors