Skip to content
This repository was archived by the owner on Jul 10, 2025. It is now read-only.

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 ([email protected])
Sponsor Faizan Muhammad ([email protected])
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 subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant