Searched refs:keys_multi_index (Results 1 – 1 of 1) sorted by relevance
191 std::vector<llvm::Value*> keys_multi_index = tiled_keys_index.multidim(); in EmitTiledCompareLoop() local194 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()195 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()273 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()274 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop()357 std::vector<llvm::Value*> keys_multi_index(rank); in EmitSortInPlace() local359 keys_multi_index[iteration_order_to_logical_order[i]] = tiles_index[i]; in EmitSortInPlace()362 IrArray::Index keys_index(keys_multi_index, values_arrays[0].GetShape(), in EmitSortInPlace()370 keys_multi_index[dimension_to_sort] = index; in EmitSortInPlace()371 IrArray::Index keys_index(keys_multi_index, in EmitSortInPlace()[all …]