[Relay][ONNX] fix bug in from_onnx (#2430)
authorhlu1 <14827759+hlu1@users.noreply.github.com>
Tue, 15 Jan 2019 05:15:03 +0000 (21:15 -0800)
committerTianqi Chen <tqchen@users.noreply.github.com>
Tue, 15 Jan 2019 05:15:03 +0000 (21:15 -0800)
tutorials/relay/from_onnx.py

index 46b803a3a7835364da3190909922a7fe42e8b81f..90b0c1c6645ca74067e45e83124c2452943eb966 100644 (file)
@@ -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()
 
 ######################################################################