torch-mlir

Commit Graph

Author	SHA1	Message	Date
Aaron Arthurs	85898aaf10	Add TCF convolutional op with bias addition (#137 )	2020-12-15 12:53:12 -08:00
Sean Silva	32388d938b	Make some passes run on FuncOp so they can run in parallel.	2020-11-13 16:12:18 -08:00
Sean Silva	1c7c362e29	[TCP] Replace tcp.matmul with linalg.matmul. This involved adding a `tcp.splatted` op to splat a dynamically sized init tensor. See rationale in TCPOps.td docs. One interesting observation is that when lowering tcf.matmul to linalg.matmul, we need to both 1) create the error checks and 2) calculate a shape transfer function to create the init tensors. Previously, 2) was deferred to bufferizing tcp.matmul later. I'm not sure if this is a conflation of concerns or not. For now, it's not a big burden.	2020-11-10 18:58:28 -08:00

Author

SHA1

Message

Date

Aaron Arthurs

85898aaf10

Add TCF convolutional op with bias addition (#137 )

2020-12-15 12:53:12 -08:00

Sean Silva

32388d938b

Make some passes run on FuncOp so they can run in parallel.

2020-11-13 16:12:18 -08:00

Sean Silva

1c7c362e29

[TCP] Replace tcp.matmul with linalg.matmul.

This involved adding a `tcp.splatted` op to splat a dynamically sized
init tensor. See rationale in TCPOps.td docs.

One interesting observation is that when lowering tcf.matmul to
linalg.matmul, we need to both 1) create the error checks and 2)
calculate a shape transfer function to create the init tensors.
Previously, 2) was deferred to bufferizing tcp.matmul later. I'm not
sure if this is a conflation of concerns or not. For now, it's not a big
burden.

2020-11-10 18:58:28 -08:00

3 Commits (d8261a06d56df47210dc67e6d3a4fd8a0400165f)