添加 BATCH_TO_SPACE_ND 和 SPACE_TO_BATCH_ND 到 tensorflow lite

Question

NNAPI 1.1 已经支持 SPACE_TO_BATCH_ND 和 BATCH_TO_SPACE_ND 操作，但当前的 TFLite 代码尚未实现。参见：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/nnapi_delegate.cc#L607

我试过这样添加它们：

case tflite::BuiltinOperator_SPACE_TO_BATCH_ND:
   nn_op_type = ANEURALNETWORKS_SPACE_TO_BATCH_ND;
   break;
case tflite::BuiltinOperator_BATCH_TO_SPACE_ND:
   nn_op_type = ANEURALNETWORKS_BATCH_TO_SPACE_ND;
   break;

但它不起作用，当我在 android 设备上运行它时出现错误消息：E/Utils：输入操作数无效运行ds (3, expected 2) or output ope运行ds (1, expected 1) for operation BATCH_TO_SPACE_ND

如何正确地将这些操作添加到 tensorflow lite？谢谢！

Answer 1

找到答案，要做到这一点，你得花点时间了解他们的代码架构，然后尝试"translate"从tftensor到NNAPI：

例如：ANEURALNETWORKS_BATCH_TO_SPACE_ND (NNAPI) 需要 2 个参数（输入张量，block_shape），而 TFLite 需要 3 个参数（输入张量，block_shape，作物），所以你需要处理它以绕过错误。

Answer 2

前面的回答是正确的。当时 NNAPI 不支持裁剪，因此代码仅委托运算符 with zero crops.

添加 BATCH_TO_SPACE_ND 和 SPACE_TO_BATCH_ND 到 tensorflow lite

Adding BATCH_TO_SPACE_ND and SPACE_TO_BATCH_ND to tensorflow lite

c++

android

tensorflow

nnapi