Skip to content

Commit a595ac8

Browse files
author
Nikhil Thorat
authored
Remove gradMode from public API as it is confusing users. (tensorflow#1414)
1 parent f217045 commit a595ac8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/environment.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,8 +182,8 @@ export class Environment {
182182
*/
183183
/** @doc {heading: 'Performance', subheading: 'Memory'} */
184184
static tidy<T extends TensorContainer>(
185-
nameOrFn: string|ScopeFn<T>, fn?: ScopeFn<T>, gradMode = false): T {
186-
return ENV.engine.tidy(nameOrFn, fn, gradMode);
185+
nameOrFn: string|ScopeFn<T>, fn?: ScopeFn<T>): T {
186+
return ENV.engine.tidy(nameOrFn, fn);
187187
}
188188

189189
/**

0 commit comments

Comments
 (0)