File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/python/frontend/tflite Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1616from tensorflow .python .ops import variables
1717from tensorflow .contrib .lite .python import interpreter as interpreter_wrapper
1818
19- import nnvm .testing .tf
19+ import relay .testing .tf
2020
2121#######################################################################
2222# Generic run functions for TVM & TFLite
@@ -344,7 +344,7 @@ def test_forward_mobilenet():
344344 '''test mobilenet v1 tflite model'''
345345 # MobilenetV1
346346 temp = util .tempdir ()
347- tflite_model_file = nnvm .testing .tf .get_workload_official (
347+ tflite_model_file = relay .testing .tf .get_workload_official (
348348 "http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz" ,
349349 "mobilenet_v1_1.0_224.tflite" , temp )
350350 tflite_model_buf = open (tflite_model_file , "rb" ).read ()
You can’t perform that action at this time.
0 commit comments