Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Namespaces | Functions
dragon::utils Namespace Reference

Namespaces

 math
 

Functions

void IncreaseIndexInDims (const int ndims, const int *dims, int *index)
 
bool IsRowwiseBroadcast (const vec64_t &A_dims, const vec64_t &B_dims, int *rows, int *cols)
 
bool IsColwiseBroadcast (const vec64_t &A_dims, const vec64_t &B_dims, int *rows, int *cols)
 
bool IsRowwiseReduce (const int ndims, const int *A_dims, const int *B_dims, int *rows, int *cols)
 
bool IsColwiseReduce (const int ndims, const int *A_dims, const int *B_dims, int *rows, int *cols)
 
void ComputeTransposedAxesForReduce (const int ndims, const int naxes, const int *reduce_axes, int *transpose_axes)
 
void ComputeTransposedStrides (const int ndims, const int *dims, const int *axes, int *strides)
 

Function Documentation

◆ ComputeTransposedAxesForReduce()

void dragon::utils::ComputeTransposedAxesForReduce ( const int  ndims,
const int  naxes,
const int *  reduce_axes,
int *  transpose_axes 
)

◆ ComputeTransposedStrides()

void dragon::utils::ComputeTransposedStrides ( const int  ndims,
const int *  dims,
const int *  axes,
int *  strides 
)

◆ IncreaseIndexInDims()

void dragon::utils::IncreaseIndexInDims ( const int  ndims,
const int *  dims,
int *  index 
)

◆ IsColwiseBroadcast()

bool dragon::utils::IsColwiseBroadcast ( const vec64_t A_dims,
const vec64_t B_dims,
int *  rows,
int *  cols 
)

◆ IsColwiseReduce()

bool dragon::utils::IsColwiseReduce ( const int  ndims,
const int *  A_dims,
const int *  B_dims,
int *  rows,
int *  cols 
)

◆ IsRowwiseBroadcast()

bool dragon::utils::IsRowwiseBroadcast ( const vec64_t A_dims,
const vec64_t B_dims,
int *  rows,
int *  cols 
)

◆ IsRowwiseReduce()

bool dragon::utils::IsRowwiseReduce ( const int  ndims,
const int *  A_dims,
const int *  B_dims,
int *  rows,
int *  cols 
)