добавить опцию в tenorflow, но тип это ошибка - PullRequest
0 голосов
/ 26 ноября 2018

Мне нужно добавить OP в tenorflow & lite. Код примерно такой:

REGISTER_OP("CheckSize") .Input("y: uint8")
  • если тип int32, все в порядке (сгенерируйте мою модель и перейдите в liteмодель успешно).
  • Но если тип ввода - uint8 или int8 (возможно, другой тип такой же), можно создать модель тензорного потока, но при переходе на облегченную модель информация об ошибке будет такой, как показано ниже:

    2018-11-26 09: 25: 12.366596: I tenorflow / contrib / lite / toco / import_tensorflow.cc: 1055] Преобразование неподдерживаемой операции: CheckSize 2018-11-26 09: 25: 12.366683: I тензор потока/contrib/lite/toco/import_tensorflow.cc:1055] Преобразование неподдерживаемой операции: CheckSize 2018-11-26 09: 25: 12.366702: I tenorflow / contrib / lite / toco / import_tensorflow.cc: 1055] Преобразование неподдерживаемой операции: CheckSize 2018-11-26 09: 25: 12.366737: I tenorflow / contrib / lite / toco / import_tensorflow.cc: 1055] Преобразование неподдерживаемой операции: CheckSize 2018-11-26 09: 25: 12.368419: F тензор потока / contrib / lite / toco /tooling_util.cc:967] Проверка не удалась: array-> has_shape ()

Кто-нибудь знает причину, как я могу использовать тип ввода int8?

...