[Relay][ONNX] fix bug in from_onnx (#2430)

b9604671 · hlu1 · Tianqi Chen · 3bbfc2dc · b9604671
Commit b9604671 authored Jan 14, 2019 by hlu1 Committed by Tianqi Chen Jan 14, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 0 deletions

tutorials/relay/from_onnx.py
+1 -0

No files found.
--- a/tutorials/relay/from_onnx.py
+++ b/tutorials/relay/from_onnx.py
@@ -75,6 +75,7 @@ with relay.build_config(opt_level=1):
 ######################################################################
 # Execute on TVM
 # ---------------------------------------------
+dtype = 'float32'
 tvm_output = intrp.evaluate(sym)(tvm.nd.array(x.astype(dtype)), **params).asnumpy()

 ######################################################################