tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] wyc-ruiker commented on issue #5041: [TFLITE]DepthToSpace and SpaceToDepth support
Date Fri, 20 Mar 2020 14:56:10 GMT
wyc-ruiker commented on issue #5041: [TFLITE]DepthToSpace and SpaceToDepth support
URL: https://github.com/apache/incubator-tvm/pull/5041#issuecomment-601742509
 
 
   When I run TFLite test scripts in Tensorflow r2.0, there is an error:
   
   >  File "test_forward.py", line 1778, in <module>
   >     test_forward_depthtospace()
   > 
   >   File "test_forward.py", line 1466, in test_forward_depthtospace
   >     _test_depthtospace(np.random.normal(size=[1, 32, 32, 4]).astype("float32"), 2)
   > 
   >   File "test_forward.py", line 1461, in _test_depthtospace
   >     compare_tflite_with_tvm(data, 'Placeholder:0', [in_data], [out])
   > 
   >   File "test_forward.py", line 181, in compare_tflite_with_tvm
   >     tflite_model_buffer = converter.convert()
   > 
   > Some of the operators in the model are not supported by the standard TensorFlow Lite
runtime. If those are native TensorFlow operators, you might be able to use the extended runtime
by passing --enable_select_tf_ops, or by setting target_ops=TFLITE_BUILTINS,SELECT_TF_OPS
when calling tf.lite.TFLiteConverter(). Otherwise, if you have a custom implementation for
them you can disable this error with --allow_custom_ops, or by setting allow_custom_ops=True
when calling tf.lite.TFLiteConverter(). Here is a list of builtin operators you are using:
. Here is a list of operators for which you will need custom implementations: DEPTH_TO_SPACE.
   
   But I don't know why. @siju-samuel Can you help me figure it out?

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message