向tensorflow lite添加BATCH\u TO_SPACE\u ND和SPACE\u TO_BATCH\u ND

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:BATCH\u TO\u SPACE\ND操作的输入操作数（3，预期为2）或输出操作数（1，预期为1）的数量无效

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

Answer 1

前面的答案是正确的。当时NNAPI不支持作物，所以代码只委托零作物的运算符。

Answer 2

找到了一个答案，要做到这一点，你必须花一些时间理解他们的代码架构，然后尝试将tftensor“翻译”为NNAPI：

例如：ANEURALNETWORKS\u BATCH\u TO\u SPACE\u ND（NNAPI）需要2个参数（输入张量，block\u形状），而TFLite需要3个参数（输入张量，block\u形状，crops），因此需要处理它以绕过错误。