Skip to content

Commit 66498c2

Browse files
authored
Add RematScope API to keras-io (#2080)
* add RematScope API to keras-io * add reference * add keras.remat * renaming to Rematerialization and remove from utility section * version bump * revert version
1 parent d9c4ed0 commit 66498c2

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

scripts/api_master.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1952,6 +1952,23 @@
19521952
},
19531953
],
19541954
},
1955+
{
1956+
"path": "rematerialization/",
1957+
"title": "Rematerialization",
1958+
"toc": True,
1959+
"children": [
1960+
{
1961+
"path": "remat_scope",
1962+
"title": "RematScope",
1963+
"generate": ["keras.RematScope"],
1964+
},
1965+
{
1966+
"path": "remat",
1967+
"title": "Remat",
1968+
"generate": ["keras.remat"],
1969+
},
1970+
],
1971+
},
19551972
{
19561973
"path": "utils/",
19571974
"title": "Utilities",

0 commit comments

Comments
 (0)