MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / shift_right

Method shift_right

arm_compute/core/TensorShape.h:149–158  ·  view source on GitHub ↗

Shifts right the tensor shape increasing its dimensions * * @param[in] step Rotation step */

Source from the content-addressed store, hash-verified

147 * @param[in] step Rotation step
148 */
149 void shift_right(size_t step)
150 {
151 ARM_COMPUTE_ERROR_ON(step > TensorShape::num_max_dimensions - num_dimensions());
152
153 std::rotate(begin(), begin() + TensorShape::num_max_dimensions - step, end());
154 _num_dimensions += step;
155
156 // Correct number dimensions to ignore trailing dimensions of size 1
157 apply_dimension_correction();
158 }
159
160 /** Return a copy with collapsed dimensions starting from a given point.
161 *

Callers 3

compute_col2im_shapeFunction · 0.80
compute_softmax_shapeFunction · 0.80

Calls 1

endFunction · 0.85

Tested by

no test coverage detected