mxnet
Classes | Namespaces | Functions
dot_engine-inl.h File Reference

definitions of how Matrix Multiplications can be evaluated More...

#include <vector>
#include "./base.h"
#include "./extension/implicit_gemm.h"
Include dependency graph for dot_engine-inl.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mshadow::expr::DotEngine< SV, Device, ddim, ldim, rdim, ltrans, rtrans, DType >
 
struct  mshadow::expr::BLASEngine< Device, DType >
 
struct  mshadow::expr::BLASEngine< cpu, float >
 
struct  mshadow::expr::BLASEngine< cpu, double >
 
struct  mshadow::expr::BLASEngine< gpu, half::half_t >
 
struct  mshadow::expr::BLASEngine< gpu, float >
 
struct  mshadow::expr::BLASEngine< gpu, double >
 
struct  mshadow::expr::DotEngine< SV, xpu, 2, 2, 2, transpose_left, transpose_right, DType >
 
struct  mshadow::expr::DotEngine< SV, xpu, 1, 1, 2, false, transpose_right, DType >
 
struct  mshadow::expr::DotEngine< SV, xpu, 2, 1, 1, true, false, DType >
 

Namespaces

 mshadow
 overloaded + operator between half_t and bf16_t
 
 mshadow::expr
 namespace for abstract expressions and expressions template, have no dependency 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

template<typename Device , typename DType >
void mshadow::GetBatchedView (DType **dst, DType *src, int num, int stride, Stream< Device > *stream)
 CPU/GPU: Get a batched view of the src array. dst[i] = src + i * stride. More...
 
template<typename DType >
void mshadow::GetBatchedView (DType **dst, DType *src, int num, int stride, Stream< cpu > *stream)
 
Shape< 2 > mshadow::expr::GetShape (const Shape< 2 > &shape, bool transpose)
 

Detailed Description

definitions of how Matrix Multiplications can be evaluated

Copyright (c) 2014 by Contributors

Author
Tianqi Chen