Searched refs:reverse_map (Results 1 – 4 of 4) sorted by relevance
202 uint8_t reverse_map[64] = {}; in update_so_info() local209 reverse_map[slot++] = bit; in update_so_info()215 sh->streamout.so_info_slots[i] = reverse_map[output->register_index]; in update_so_info()
750 uint8_t reverse_map[64] = {0}; in update_so_info() local754 reverse_map[slot++] = u_bit_scan64(&outputs_written); in update_so_info()760 output->register_index = reverse_map[output->register_index]; in update_so_info()
322 uint8_t reverse_map[64] = {}; in update_so_info() local325 reverse_map[slot++] = u_bit_scan64(&outputs_written); in update_so_info()332 output->register_index = reverse_map[output->register_index]; in update_so_info()
255 std::vector<int64> reverse_map(hlo->shape().rank()); in AddInstruction() local256 for (int64 i = 0; i < reverse_map.size(); ++i) { in AddInstruction()257 reverse_map[hlo->dimensions(i)] = i; in AddInstruction()260 xpose_broadcast_dims.insert(reverse_map[dim]); in AddInstruction()