blackkker commented on a change in pull request #9840:
URL: https://github.com/apache/tvm/pull/9840#discussion_r782674820
##########
File path: python/tvm/relay/frontend/tflite.py
##########
@@ -1911,6 +1911,14 @@ def convert_fully_connected(self, op):
weight_expr = self.exp_tab.new_const(weight_value,
dtype=weight_tensor_type_str)
weight_shape = _infer_shape(weight_expr)
+ # Change the output shape calculation based on keep_dim option
Review comment:
The fact is that I can not construct a fully_connected op with keep_dim
== True or find a model which fully_connected op with keep_dim == True.
Also, I have tried to construct a matmul in tensorflow which can be
converted to fullyconnected by tf.lite.TFLiteConverter. But, I just get the
FlexMatmul op in TFLite.
Now, i am stuck。
BTW, I implement it according to
[https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/kernels/fully_connected.cc](url)
line:353
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
