Skip to content

Commit ee8b2ae

Browse files
authored
core: Do not export conv gradient ops. (#1992)
DEV
1 parent 37baaf2 commit ee8b2ae

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tfjs-core/src/ops/ops.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@ export * from './batchnorm';
1919
export * from './boolean_mask';
2020
export * from './complex_ops';
2121
export * from './concat_split';
22-
export * from './conv';
22+
// Selectively exporting to avoid exposing gradient ops.
23+
export {conv1d, conv2d, conv3d, depthwiseConv2d, separableConv2d, conv2dTranspose, conv3dTranspose} from './conv';
2324
export * from './matmul';
2425
export * from './reverse';
2526
export * from './pool';

0 commit comments

Comments
 (0)