Skip to content

Commit c86a8fb

Browse files
committed
Update comment
1 parent 054ddae commit c86a8fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/backends/tensorflow.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,10 +168,10 @@ RAI_Graph *RAI_GraphCreateTF(const char *prefix,
168168
// For setting config options in session from the C API see:
169169
// https://github.com/tensorflow/tensorflow/issues/13853
170170
// import tensorflow as tf
171-
// config = tf.ConfigProto()
172-
// config.intra_op_parallelism_threads = 1
171+
// config = tf.ConfigProto(device_count = {'GPU': 0})
173172
// serialized = config.SerializeToString()
174173
// result = list(map(hex, serialized))
174+
// print(result)
175175

176176
// TODO: complain if device is GPU and GPU not available?
177177
if (device == RAI_DEVICE_CPU) {

0 commit comments

Comments
 (0)