Searched refs:tile_to_use (Results 1 – 2 of 2) sorted by relevance
69 SimpleTensor<T> *tile_to_use = lhs_info.transpose ? &src_tile_transposed : &src_tile; in gemm_reshape_lhs_matrix() local71 … int offset_output_x = lhs_info.interleave ? tile_to_use->shape()[0] : tile_to_use->shape()[0] * t… in gemm_reshape_lhs_matrix()72 …signed int step_output_x = lhs_info.interleave ? tile_to_use->shape()[0] * lhs_info.v0 : tile_to… in gemm_reshape_lhs_matrix()92 for(unsigned int i = 0; i < tile_to_use->shape()[1]; ++i) in gemm_reshape_lhs_matrix()94 const unsigned int offset_tile = i * tile_to_use->shape()[0]; in gemm_reshape_lhs_matrix()97 …std::copy(&(*tile_to_use)[offset_tile], &(*tile_to_use)[offset_tile + tile_to_use->shape()[0]], &o… in gemm_reshape_lhs_matrix()
69 SimpleTensor<T> *tile_to_use = rhs_info.transpose ? &src_tile_transposed : &src_tile; in gemm_reshape_rhs_matrix() local71 … int offset_output_x = rhs_info.interleave ? tile_to_use->shape()[0] : tile_to_use->shape()[0] * t… in gemm_reshape_rhs_matrix()72 …signed int step_output_x = rhs_info.interleave ? tile_to_use->shape()[0] * rhs_info.h0 : tile_to… in gemm_reshape_rhs_matrix()94 for(unsigned int i = 0; i < tile_to_use->shape()[1]; ++i) in gemm_reshape_rhs_matrix()96 const unsigned int offset_tile = i * tile_to_use->shape()[0]; in gemm_reshape_rhs_matrix()99 …std::copy(&(*tile_to_use)[offset_tile], &(*tile_to_use)[offset_tile + tile_to_use->shape()[0]], &o… in gemm_reshape_rhs_matrix()