Skip to content

Commit 7396669

Browse files
jugyonkreeger
authored andcommitted
Change API registration. (#88)
* Change API registration.
1 parent 27468ac commit 7396669

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

demo/src/mnist_data_demo.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,12 @@
1616
*/
1717

1818
import * as tf from '@tensorflow/tfjs-core';
19-
import {bindTensorFlowBackend} from '@tensorflow/tfjs-node';
19+
import '@tensorflow/tfjs-node';
2020

2121
import {MnistDataset} from './mnist_data';
2222
import {Timer} from './timer';
2323

24-
bindTensorFlowBackend();
24+
tf.setBackend('tensorflow');
2525

2626
function testPrint(image: tf.Tensor, label: tf.Tensor) {
2727
const data = image.dataSync();

0 commit comments

Comments
 (0)