26 #ifndef MSHADOW_EXTENSION_PAD_H_ 27 #define MSHADOW_EXTENSION_PAD_H_ 28 #include "../extension.h" 37 template<
typename SrcExp,
typename DType,
int srcdim>
40 SrcExp, srcdim, DType> {
53 : src_(src), pad_y_(pad_y), pad_x_(pad_x) {
55 src_height_ = this->
shape_[srcdim - 2];
56 src_width_ = this->
shape_[srcdim - 1];
57 this->
shape_[srcdim - 2] += pad_y * 2;
58 this->
shape_[srcdim - 1] += pad_x * 2;
70 template<
typename SrcExp,
typename DType,
int etype>
74 ::Error_Expression_Does_Not_Meet_Dimension_Req();
87 template<
typename SrcExp,
typename DType,
int etype>
91 ::Error_Expression_Does_Not_Meet_Dimension_Req();
93 (src.
self(), pad_y, pad_x);
98 template<
typename SrcExp,
typename DType,
int srcdim>
104 new_height_(e.
shape_[srcdim - 2]),
108 const index_t y = i % new_height_;
109 const index_t c = i / new_height_;
110 if (y <
pad_y_ || x <
pad_x_)
return static_cast<DType
>(0);
116 return static_cast<DType
>(0);
130 #endif // MSHADOW_EXTENSION_PAD_H_ const SrcExp & src_
source operand
Definition: pad.h:42
PaddingExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > pad(const Exp< SrcExp, DType, etype > &src, index_t pad)
padding expression, pad a image with zeros on boundaries, padding affects shape[0], and shape[1]
Definition: pad.h:72
Definition: expr_engine-inl.h:59
used to help static type check
Definition: expr_engine-inl.h:331
index_t pad_y_
pad size in y
Definition: pad.h:44
padding expression, pad a image with zeros
Definition: pad.h:38
static Shape< dim > Check(const E &t)
#define MSHADOW_XINLINE
Definition: base.h:223
MSHADOW_XINLINE DType Eval(index_t i, index_t j) const
Definition: pad.h:106
int32_t index_t
type that will be used for index
Definition: base.h:336
index_t src_height_
source tensor height
Definition: pad.h:48
index_t src_width_
source tensor width
Definition: pad.h:50
index_t pad_x_
pad size in x
Definition: pad.h:46
defines how expression exp can be evaluated and stored into dst
Definition: expression.h:80
const SubType & self(void) const
Definition: expression.h:83
Plan< BinaryMapExp< OP, TA, TB, DType, etype >, DType > MakePlan(const BinaryMapExp< OP, TA, TB, DType, etype > &e)
Definition: expr_engine-inl.h:240
a general class that allows extension that makes tensors of some shape
Definition: expr_engine-inl.h:44
overloaded + operator between half_t and bf16_t
Definition: base.h:327
Plan(const PaddingExp< SrcExp, DType, srcdim > &e)
Definition: pad.h:101
Shape< dim > shape_
the shape of this expression
Definition: expr_engine-inl.h:48
PaddingExp(const SrcExp &src, index_t pad_y, index_t pad_x)
constructor
Definition: pad.h:52