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

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 "nnvm/op_attr_types.h"
#include "nnvm/graph_attr_types.h"
#include "nnvm/graph.h"
#include "nnvm/pass.h"
Include dependency graph for tuple.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  mxnet::Tuple< ValueType >
 A dynamic sized array data structure that is optimized for storing small number of elements with same type. More...
 
class  mxnet::TShape
 A Shape class that is used to represent shape of each tensor. More...
 
struct  std::hash< mxnet::Tuple< T > >
 hash function for Tuple. More...
 
struct  std::hash< mxnet::TShape >
 hash function for TShape. More...
 
struct  dmlc::type_name_helper< mxnet::Tuple< T > >
 

Namespaces

 mxnet
 namespace of mxnet
 
 std
 
 dmlc
 namespace for dmlc
 

Typedefs

using mxnet::ShapeVector = std::vector< mxnet::TShape >
 The result holder of shape of each NodeEntry in the graph. More...
 
using mxnet::FInferShape = nnvm::FInferNodeEntryAttr< mxnet::TShape >
 Shape inference function. Update the shapes given the input shape information. TShape.ndim() == -1 means the shape is still unknown. More...
 

Functions

bool mxnet::ndim_is_known (const int ndim)
 
bool mxnet::dim_size_is_known (const dim_t dim_size)
 
bool mxnet::ndim_is_known (const TShape &x)
 
bool mxnet::dim_size_is_known (const TShape &x, const int idx)
 
bool mxnet::shape_is_known (const TShape &x)
 
bool mxnet::shape_is_known (const std::vector< TShape > &shapes)
 
template<typename SrcIter , typename DstIter >
DstIter mxnet::ShapeTypeCast (const SrcIter begin, const SrcIter end, DstIter dst_begin)
 helper function to cast type of container elements More...
 
template<typename SrcIter >
TShape mxnet::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< mxnet::TShape >,"Shape or None")
 description for optional TShape More...
 
 dmlc::DMLC_DECLARE_TYPE_NAME (optional< mxnet::Tuple< int >>,"Shape or None")
 

Detailed Description

Data structure Tuple and TShape to store dynamic sized shapes.

Copyright (c) 2019 by Contributors