mxnet
|
Data structure Tuple and TShape to store dynamic sized shapes. More...
#include <vector>
#include <type_traits>
#include <algorithm>
#include <utility>
#include <iostream>
#include <string>
#include "base.h"
Go to the source code of this file.
Classes | |
class | nnvm::Tuple< ValueType > |
A dynamic sized array data structure that is optimized for storing small number of elements with same type. More... | |
class | nnvm::TShape |
A Shape class that is used to represent shape of each tensor. More... | |
struct | std::hash< nnvm::Tuple< T > > |
hash function for Tuple. More... | |
struct | std::hash< nnvm::TShape > |
hash function for TShape. More... | |
struct | dmlc::type_name_helper< nnvm::Tuple< T > > |
Namespaces | |
nnvm | |
std | |
dmlc | |
namespace for dmlc | |
Typedefs | |
typedef int64_t | nnvm::dim_t |
data type to store dim size More... | |
Functions | |
template<typename SrcIter , typename DstIter > | |
DstIter | nnvm::ShapeTypeCast (const SrcIter begin, const SrcIter end, DstIter dst_begin) |
helper function to cast type of container elements More... | |
template<typename SrcIter > | |
TShape | nnvm::ShapeTypeCast (const SrcIter begin, const SrcIter end) |
helper function to transform a container to TShape with type cast More... | |
dmlc::DMLC_DECLARE_TYPE_NAME (optional< nnvm::TShape >,"Shape or None") | |
description for optional TShape More... | |
Data structure Tuple and TShape to store dynamic sized shapes.