xla
xla copied to clipboard
Fold MapOp to one of its ContractionOp operands
Fold MapOp to one of its ContractionOp operands
Given a MapOp that adds a ContractionOp to some other op A, fold the MapOp by making A the init operand of the ContractionOp.