[GitHub] [incubator-tvm] wyc-ruiker commented on issue #5041: [TFLITE]DepthToSpace and SpaceToDepth support

[GitBox <gi...@apache.org>]

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