Skip to content

Commit 7fa284d

Browse files
authored
a minor fix in finding unused ops algo (tf-coreml#183)
1 parent 04ca753 commit 7fa284d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tfcoreml/_tf_coreml_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def _convert_pb_to_mlmodel(tf_model_path,
206206
input_feed_dict2[op.outputs[0]] = 1
207207
else:
208208
input_feed_dict[op.outputs[0]] = np.random.rand(*shape)
209-
input_feed_dict2[op.outputs[0]] = np.random.rand(*shape)
209+
input_feed_dict2[op.outputs[0]] = 255*np.random.rand(*shape)
210210

211211
SHAPE_DICT[input_name] = shape
212212

0 commit comments

Comments
 (0)