Searched refs:col_id (Results 1 – 2 of 2) sorted by relevance
299 size_t col_id = column_name_id_map_[itr.first]; in MapColumns() local301 (*out_q_table)[i][col_id] = std::move((*in_q_table)[i][itr.second]); in MapColumns()310 size_t col_id = column_name_id_map_[out_col_names_[i]]; in MapColumns() local317 t_row[col_id] = out_cols[i][row_id++]; in MapColumns()442 for (size_t col_id : pad_cols) { in PadColumns() local443 max_shapes[col_id] = std::vector<dsize_t>((*table)->front()[col_id]->Rank(), -1); in PadColumns()444 …if (pad_shapes[col_id].empty()) pad_shapes[col_id] = max_shapes[col_id]; // fill pad shape with -1 in PadColumns()446 pad_shapes[col_id].size() == max_shapes[col_id].size(), in PadColumns()448 …std::to_string(pad_shapes[col_id].size()) + ", column rank: " + std::to_string(max_shapes[col_id].… in PadColumns()453 for (size_t col_id : pad_cols) { // iterator each tensor in a row in PadColumns() local[all …]
159 for (int col_id = shm_col_id; col_id < (TileWidth); in Swap3DTensorLast2DimKernel_shared() local160 col_id += NumColsPerWriteLoop) { // move to the next pass, loop in Swap3DTensorLast2DimKernel_shared()162 … output[output_idx] = shm_tile[shm_row_id * (TileWidth + 1) + col_id]; // avoid bank conflict in Swap3DTensorLast2DimKernel_shared()167 for (int col_id = shm_col_id; col_id < (tile_width); col_id += NumColsPerWriteLoop) { in Swap3DTensorLast2DimKernel_shared() local169 output[output_idx] = shm_tile[shm_row_id * (TileWidth + 1) + col_id]; in Swap3DTensorLast2DimKernel_shared()