[RFC] CompositeTensor Layering for tf.function

[Umer Javed]

Hi everyone,

I am happy to announce that RFC: CompositeTensor Layering for tf.function is available for comment.

The feedback phase will be open for two weeks until 2022-12-01. Comments are invited to the pull request linked. You can view the design doc here and can leave comments inline on the document source.

Summary:

• This RFC introduces a solution for abstracting away CompositeTensor and dependencies to it away from tf.function by moving the dependency towards TraceType
• This RFC is a part of the general tf.function layering effort
• CompositeTensors are of TraceType TypeSpec. Their logic for tracing can be handled by their TraceType.
• Placeholders, Graph IOs, and casting logic can be abstracted to TraceType.
• Impact-wise, this opens up opportunities to further layer tf.function, introduce new Tensor APIs without having to configure tracing logic, and make changes to tracing logic without having to handle CompositeTensor cases uniquely.

Thank you for your review and participation!

Umer Javed