Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RFC: CompositeTensor Layering for tf.function #437

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

u7javed
Copy link

@u7javed u7javed commented Nov 17, 2022

This RFC will be open for comment until Thursday, December 1st, 2022.

CompositeTensor Layering for tf.function

Status Proposed
RFC # 437
Author(s) Umer Javed (ujaved@google.com)
Sponsor Faizan Muhammad (fmuham@google.com)
Updated 2022-11-17

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

  • 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
  • Basic implementation proposals with code snippets

@u7javed u7javed marked this pull request as ready for review November 17, 2022 23:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant