Dimension --> <1, 1, 1, A*B*C*D>
| 49 | |
| 50 | // Dimension <A, B, C, D> --> <1, 1, 1, A*B*C*D> |
| 51 | vector<int> flattenTensorDim (vector<int>& src) |
| 52 | { |
| 53 | int finalDim = 1; |
| 54 | for(int i=0; i<src.size(); i++) |
| 55 | { |
| 56 | finalDim *= src.at(i); |
| 57 | } |
| 58 | |
| 59 | vector<int> dst(4, 1); // 1x1x1x1 |
| 60 | dst.at(3) = finalDim; // 1x1x1xfinalDim |
| 61 | return dst; |
| 62 | } |
| 63 | |
| 64 | // Get total number of elems in a blob |
| 65 | int getSize(vector<int>& src) |
no test coverage detected