Searched refs:chunk_width (Results 1 – 3 of 3) sorted by relevance
70 RawTensor transpose(const RawTensor &src, int chunk_width) in transpose() argument74 dst_shape.set(0, src.shape().y() * chunk_width); in transpose()75 dst_shape.set(1, std::ceil(src.shape().x() / static_cast<float>(chunk_width))); in transpose()82 for(int i = 0; i < dst.num_elements(); i += chunk_width) in transpose()86 coord.set(0, coord.y() * chunk_width); in transpose()87 coord.set(1, coord_x / chunk_width); in transpose()89 const int num_elements = std::min<int>(chunk_width, src.shape().x() - coord.x()); in transpose()93 out_ptr += chunk_width * dst.element_size(); in transpose()
123 RawTensor transpose(const RawTensor &src, int chunk_width = 1);
7480 int chunk_width = msb - lsb + 1; in SubstituteIntField() local7481 VIXL_ASSERT((chunk_width > 0) && (chunk_width < 32)); in SubstituteIntField()7482 bits = (bits << chunk_width) | (instr->ExtractBits(msb, lsb)); in SubstituteIntField()7483 width += chunk_width; in SubstituteIntField()