| 546 | } |
| 547 | |
| 548 | af::dim4 calcStrides(const af::dim4 &parentDim) { |
| 549 | af::dim4 out(1, 1, 1, 1); |
| 550 | dim_t *out_dims = out.get(); |
| 551 | const dim_t *parent_dims = parentDim.get(); |
| 552 | |
| 553 | for (dim_t i = 1; i < 4; i++) { |
| 554 | out_dims[i] = out_dims[i - 1] * parent_dims[i - 1]; |
| 555 | } |
| 556 | |
| 557 | return out; |
| 558 | } |
| 559 | |
| 560 | std::string minimalDim4(af::dim4 coords, af::dim4 dims) { |
| 561 | std::ostringstream os; |