Commit 396a8a41 authored by Mark Heffernan's avatar Mark Heffernan Committed by TensorFlower Gardener
Browse files

Add custom call with layout constraints. 

Add a variant of CustomCall which specifies arbitrary layout constraints on the operands and result. The existing non-layout-constrained CustomCall is changed to have no layout preference and can now be assigned arbitrary layouts by layout assignment.

PiperOrigin-RevId: 216249615
parent bc5635dc
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment