This document proposes updates to the tf.function tracing API for layering CompositeTensor and removing dependencies on CompositeTensor for tf.function and its components. It is a part of the tf.function layering effort to reduce the complexity of tf.function and abstract various logic away from tf.function to simplify its mental model.
Scope of this document
Updates to tf.function, TraceType, and FuncGraph to Support the layering efforts for CompositeTensors
A fluid migration from current code to expected changes from proposed solution
This RFC will be open for comment until Thursday, December 1st, 2022.
CompositeTensor Layering for tf.function
Objective
This document proposes updates to the tf.function tracing API for layering CompositeTensor and removing dependencies on CompositeTensor for tf.function and its components. It is a part of the tf.function layering effort to reduce the complexity of tf.function and abstract various logic away from tf.function to simplify its mental model.
Scope of this document