Support for extra layers in TF lite for micro

Eston Z

unread,

Apr 21, 2022, 5:04:24 PM4/21/22

to SIG Micro

Hi guys,

May I ask that is it possible to transfer tf model(conv, pooling, batchnormalization and dropout and flatten layer) to tf lite for micro model? For what I can check in " all_ops_resolver.cc ", I can not find batch and dropout layers support.

Best Wishes

Yuxuan

Eston Z

unread,

Apr 21, 2022, 5:06:59 PM4/21/22

to SIG Micro, Eston Z

Or i could use some self-define operations?

Best Wishes

Yuxuan Zhang

Deqiang Chen

unread,

Apr 25, 2022, 1:42:56 PM4/25/22

to Eston Z, SIG Micro

Hello,

By "self-define" operations, I think you are referring to "custom" Op. Custom Op gives your abilities to define your own operation.

An example of custom OP is

https://github.com/tensorflow/tflite-micro/blob/main/tensorflow/lite/micro/kernels/circular_buffer.cc.

Best regards!

Deqiang

--
You received this message because you are subscribed to the Google Groups "SIG Micro" group.
To unsubscribe from this group and stop receiving emails from it, send an email to micro+un...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/micro/1ef8787d-1e02-4fab-bc44-b30cd2fc5447n%40tensorflow.org.

Reply all

Reply to author

Forward