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

Conversation

@texasmichelle
Copy link
Member

@texasmichelle texasmichelle commented Aug 28, 2020

Supports model summaries by adding an additional protocol requirement to Module and Layer. This enables the addition of tensor annotations within callAsFunction() and lays the groundwork for additional hooks.

Full explanation: design doc.

  • Add forward() protocol requirement to Module and Layer
  • Add default implementations of forward() and callAsFunction() that add annotations to output tensors
  • Add a link to the design doc
  • Update all native layer types except Recurrent
  • Add summary formatting

@texasmichelle texasmichelle marked this pull request as ready for review September 3, 2020 03:55
@texasmichelle texasmichelle merged commit a2fa8c4 into tensorflow:master Sep 9, 2020
@texasmichelle texasmichelle deleted the annotations2 branch September 9, 2020 17:01
@8bitmp3
Copy link
Contributor

8bitmp3 commented Sep 9, 2020

This is super helpful, thanks @texasmichelle

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.

4 participants