torch.fx.experimental.merge_matmul.merge_matmul#
- torch.fx.experimental.merge_matmul.merge_matmul(in_mod)[source]#
A graph transformation that merges matrix multiplication operations that share the same right-hand side operand into one large matrix multiplication.
____ _________ _________ ---- | | | | M| A * C | M| A | T| B | * K| C | = |---------| ---- , | | | | T| B * C | K ---- --------- --------- K R R
