Include dependency graph for broadcast_with_axis.h:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Classes | |
| struct | mshadow::expr::BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst > |
| Broadcasting the tensor in the given axis. If keepdim is off, insert the broadcasting dim after axis. Otherwise broadcasting axis. More... | |
| struct | mshadow::expr::BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc > |
| Broadcasting the tensor in multiple axes. The dimension of the source tensor in the given axes must be 1. More... | |
| struct | mshadow::expr::Plan< BroadcastWithAxisExp< SrcExp, DType, dimsrc, dimdst >, DType > |
| struct | mshadow::expr::Plan< BroadcastWithMultiAxesExp< SrcExp, DType, dimsrc >, 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 SrcExp , typename DType , int etype> | |
| BroadcastWithAxisExp< SrcExp, DType, ExpInfo< SrcExp >::kDim, ExpInfo< SrcExp >::kDim+1 > | mshadow::expr::broadcast_with_axis (const Exp< SrcExp, DType, etype > &src, const int axis, const index_t size) |
| Broadcasting the tensor after given axis. More... | |
| template<typename SrcExp , typename DType , int etype> | |
| BroadcastWithAxisExp< SrcExp, DType, ExpInfo< SrcExp >::kDim, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_keepdim (const Exp< SrcExp, DType, etype > &src, const int axis, const index_t size) |
| Broadcasting the tensor in the given axis (keepdim turned on) More... | |
| template<typename SrcExp , typename DType , int etype, typename TShape > | |
| BroadcastWithMultiAxesExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_multi_axes (const Exp< SrcExp, DType, etype > &src, const TShape &axes, const TShape &sizes) |
| Broadcasting the tensor in the given axis (keepdim turned on) More... | |
| template<typename SrcExp , typename DType , int etype, typename TShape > | |
| BroadcastWithMultiAxesExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > | mshadow::expr::broadcast_to (const Exp< SrcExp, DType, etype > &src, const TShape &target_shape) |
| Broadcasting the tensor to the target shape, dimension of different sizes must be 1 in the original tensor. More... | |
Detailed Description
Copyright (c) 2016 by Contributors
