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: [email protected] With regards, Apache Git Services
