[cleanup] When the access indices are affine exprs, rewrite assign layout using a linalg.generic

[asraa]

heir/lib/Transforms/ConvertToCiphertextSemantics/AssignLayout.cpp

Line 431 in 8f23ac0

FailureOr<Value> implementAssignLayoutNew(

Instead of nested for loops - which will make the code nicer to read and more succinct and probably more easily canonicalized.