Merged
Merged
Conversation
| in.eval(); | ||
| const dim4 idims = in.dims(); | ||
| const dim4 odims(ox, oy, idims[2], idims[3]); | ||
| out = createValueArray<T>(odims, scalar<T>(0)); |
| const dim_t sx, const dim_t sy, | ||
| const dim_t px, const dim_t py, | ||
| const bool is_column) { | ||
| if (out == 0) return AF_ERR_ARG; |
|
|
||
| dim_t nx = (ox + 2 * px - wx) / sx + 1; | ||
| dim_t ny = (oy + 2 * py - wy) / sy + 1; | ||
| ARG_ASSERT(4, wx > 0); |
| AF_ERR_ARG); | ||
| } | ||
|
|
||
| class ArgDim { |
| public: | ||
| WindowDims *wc_; | ||
| StrideDims *sc_; | ||
| PadDims *pc_; |
| af_array output; | ||
| if (allocate_out) { *out = createHandle(out_dims, in_type); } | ||
|
|
||
| DIM_ASSERT(1, getInfo(*out).dims() == out_dims); |
| af::dim4 istrides = in.strides(); | ||
| af::dim4 ostrides = out.strides(); | ||
|
|
||
| std::fill(begin(out), end(out), scalar<T>(0)); |
umar456 previously approved these changes Aug 29, 2019
Minor cleanup.
Mark Poscablo added 8 commits
February 7, 2020 16:33
Labels
None yet