Apache SINGA
A distributed deep learning platform .
|
support of sse2 optimization of some operations More...
Go to the source code of this file.
Namespaces | |
mshadow | |
namespace for mshadow | |
mshadow::sse2 | |
namespace to support sse2 vectorization | |
mshadow::expr | |
namespace for abstract expressions and expressions template, have no dependecy on tensor.h, These data structure takes no charge in computations, they are only used to define operations and represent expression in a symbolic way | |
Functions | |
void * | mshadow::sse2::AlignedMallocPitch (size_t &pitch, size_t lspace, size_t num_line) |
analog to cudaMallocPitch, allocate a aligned space with num_line * lspace cells More... | |
void | mshadow::sse2::AlignedFree (void *ptr) |
free aligned space More... | |
bool | mshadow::sse2::CheckAlign (size_t pitch) |
check if a pointer is aligned | |
bool | mshadow::sse2::CheckAlign (void *ptr) |
check if a pointer is aligned | |
index_t | mshadow::sse2::UpperAlign (index_t size, size_t fsize) |
get upper bound of aligned index of size More... | |
index_t | mshadow::sse2::LowerAlign (index_t size, size_t fsize) |
get lower bound of aligned index of size More... | |
template<typename OP , typename TA , typename TB , int etype> | |
SSEPlan< BinaryMapExp< OP, TA, TB, etype > > | mshadow::expr::MakeSSEPlan (const BinaryMapExp< OP, TA, TB, etype > &e) |
SSEPlan< ScalarExp > | mshadow::expr::MakeSSEPlan (const ScalarExp &e) |
template<typename T > | |
SSEPlan< T > | mshadow::expr::MakeSSEPlan (const ContainerExp< T > &e) |
template<typename T , int dim> | |
SSEPlan< T > | mshadow::expr::MakeSSEPlan (const MakeTensorExp< T, cpu, dim > &e) |
template<typename OP , typename TA , int etype> | |
SSEPlan< UnaryMapExp< OP, TA, etype > > | mshadow::expr::MakeSSEPlan (const UnaryMapExp< OP, TA, etype > &e) |
template<typename SV , typename E , int dim> | |
void | mshadow::MapSSEPlan (Tensor< cpu, dim > _dst, const expr::SSEPlan< E > &plan) |
use SSEPlan to compute result | |
support of sse2 optimization of some operations